Ë
    ñuki  ã                   ó”   — d dl mZmZmZmZ d dl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)é    )Ú
checkpointÚcheckpoint_policiesÚcheckpoint_nameÚprint_saved_residuals)Ú	RecomputeÚSaveableÚOffloadablezGjax.ad_checkpoint.checkpoint is deprecated; use jax.checkpoint instead.z=jax.ad_checkpoint.remat is deprecated; use jax.remat instead.)r   ÚrematN)Údeprecation_getattr)Újax._src.ad_checkpointr   Ú_deprecated_checkpointr   r   r   Ú"jax._src.interpreters.partial_evalr   r   r	   Ú_deprecationsÚtypingÚ_typingÚTYPE_CHECKINGr
   Újax._src.deprecationsr   Ú_deprecation_getattrÚ__name__Ú__getattr__© ó    úL/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/ad_checkpoint.pyú<module>r      sw   ð÷ó ÷ñ ð NØðð
 DØðñ
€ó Ø
×ÒØ%€*Ø
 €%ð
 Ùõ	 PÙ$ X¨}Ó=€+ØØÙr   