Ë
    ñukiÀ  ã                   ó˜   — d dl mZ d dl mZ ej                  dd„«       ZddefiZd dlZej                  r	 [yd dlm	Z
  e
ee«      Z[
[[y)	é    )Útraceback_util)Ú	shard_mapc                 óH   — t        j                  | |||t        «       |¬«      S )zYPlease use `jax.shard_map`. `jax.experimental.shard_map.shard_map`
  has been deprecated.)ÚmeshÚin_specsÚ	out_specsÚ
axis_namesÚ	check_vma)ÚjshmapÚ
_shard_mapÚset)Úfr   r   r   Ú	check_reps        úU/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/experimental/shard_map.pyr   r      s(   € ô 
×	Ñ	˜1 4°(ÀiÜ&)£e°yô
Bð Bó    r   zOjax.experimental.shard_map is deprecated in v0.8.0. Used jax.shard_map instead.N)Údeprecation_getattr)T)Újax._srcr   r   r   Úapi_boundaryÚ_deprecationsÚtypingÚTYPE_CHECKINGÚjax._src.deprecationsr   Ú_deprecation_getattrÚ__name__Ú__getattr__© r   r   ú<module>r      so   ðõ $Ý (à×ÑòBó ðBð ØWØðð€ó Ø	×ÒØñ õ	 PÙ$ X¨}Ó=€+ØØÙ
r   