Ë
    ñukiu  ã                   óÆ   — d dl mZ d dl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mZmZ d dlmZmZmZ ddefiZd dlZej2                  reZ[[yd d	lmZ  eee«      Z[[[y)
é    )ÚSharding)ÚNamedShardingÚSingleDeviceShardingÚPmapShardingÚset_meshÚget_mesh)ÚPartitionSpec)ÚMeshÚAbstractDeviceÚAbstractMeshÚAxisTypeÚget_abstract_meshÚuse_abstract_mesh)ÚreshardÚ	auto_axesÚexplicit_axesr   zPjax.sharding.PmapSharding is deprecated; use jax.sharding.NamedSharding instead.N)Údeprecation_getattr)Újax._src.shardingr   Újax._src.sharding_implsr   r   r   Ú_deprecated_PmapShardingr   r   Újax._src.partition_specr	   Újax._src.meshr
   r   r   r   r   r   Újax._src.pjitr   r   r   Ú_deprecationsÚtypingÚ_typingÚTYPE_CHECKINGÚjax._src.deprecationsr   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úG/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/sharding.pyú<module>r%      sz   ðõ$ 3÷õ õ÷÷ ÷ñ ð ØVØðð€ó Ø
×ÒØ)€,ð
 Ùõ	 PÙ$ X¨}Ó=€+ØØÙr#   