Ë
    Ž”vh  ã                   ó¸   — d dl mZ d dlmZmZmZmZmZ	m
Z
mZ d dlmZ d dlmZmZmZmZ de	fdefdœZd dlZej*                  re	ZeZ[yd d	lmZ  eee«      Z[[y)
é    )ÚSharding)ÚNamedShardingÚSingleDeviceShardingÚPmapShardingÚGSPMDShardingÚPositionalShardingÚuse_meshÚset_mesh)ÚPartitionSpec)ÚMeshÚAbstractMeshÚAxisTypeÚget_abstract_meshzMjax.sharding.PositionalSharding is deprecated. Use jax.NamedSharding instead.zHjax.sharding.GSPMDSharding is deprecated. Use jax.NamedSharding instead.)r   r   N)Údeprecation_getattr)Újax._src.shardingr   Újax._src.sharding_implsr   r   r   r   Ú_deprecated_GSPMDShardingr   Ú_deprecated_PositionalShardingr	   r
   Újax._src.partition_specr   Újax._src.meshr   r   r   r   Ú_deprecationsÚtypingÚTYPE_CHECKINGÚjax._src.deprecationsr   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úG/opt/face_recognition/venv/lib/python3.12/site-packages/jax/sharding.pyú<module>r!      sƒ   ðõ$ 3÷÷ ñ õ÷ó ð*ð 	'ðð*ð 	"ðñ€ó$ Ø	×ÒØ5ÐØ+€-ñ
 õ PÙ$ X¨}Ó=€+ØÙ
r   