a
    :ghÏ  ã                   @   s‚   d dl mZ d dlmZmZmZmZmZm	Z	 d dl
mZ d dlmZ ddefiZd dlZejrdeZnd dlmZ eeeƒZ[[dS )	é    )ÚSharding)ÚXLACompatibleShardingÚNamedShardingÚSingleDeviceShardingÚPmapShardingÚGSPMDShardingÚPositionalSharding)ÚPartitionSpec)ÚMeshr   zTjax.sharding.XLACompatibleSharding is deprecated. Use jax.sharding.Sharding instead.N)Údeprecation_getattr)Zjax._src.shardingr   Zjax._src.sharding_implsr   Z!_deprecated_XLACompatibleShardingr   r   r   r   r   Zjax._src.partition_specr	   Zjax._src.interpreters.pxlar
   Z_deprecationsÚtypingÚTYPE_CHECKINGZjax._src.deprecationsr   Z_deprecation_getattrÚ__name__Ú__getattr__© r   r   ú\/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/jax/sharding.pyÚ<module>   s    ûþ
