+
    –°³jc  ã                  ó  € R t ^ RIHt ^ RIt^ RIt^ RIHtHt ^ RIH	u H
t
 ^ RIH	u Ht ^ RIH	u Ht ^ RIHt ]'       d   ^ RIHt ^ RIHt ^ RIHtHt R1R R	 lltR2R
 R lltR3R R lltR4RR/R R llltR5R R lltR5R R lltRR/R R lltRRRR/R R llt R6RRRRRR/R R lllt!R5RR/R  R! lllt"R6R" R# llt#R$R%/R& R' llt$R7R( R) lltR7R* R+ llt%R, R- lt&]! ]]
P8                  R.4       ]! ]]
P:                  4       ]! ]]
P<                  4       ]! ]4       ]! ] 4       ]! ]!4       ]! ]"4       ]! ]#4       ]! ]$4       ]! ]]P                  R/4       ]! ]%]P                  R/4       ]! ]&]PL                  R04       R# )8a$  
The APIs in this file are exposed as `functorch.*`. They are thin wrappers
around the torch.func.* APIs that have deprecation warnings -- we're trying
to move people to the torch.func.* equivalents.

NB: We don't use *args, **kwargs in the signatures because that changes the
documentation.
)ÚannotationsN)ÚAnyÚTYPE_CHECKING)ÚCallable)Ú	argnums_t)Ú	in_dims_tÚ
out_dims_tFc               ó(   € V ^8„  d   QhRRRRRRRR/# )é   ÚapiÚstrÚnew_apiú
str | NoneÚreplace_newlinesÚboolÚreturn© )Úformats   "Ú\/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_functorch/deprecated.pyÚ__annotate__r      s,   € ÷ ñ Ø	ðØ!ðØ<@ðàñó    c                ó^   € Vf   RV  2pRV  RV R2pV'       d   VP                  RR4      pV# )Nztorch.func.z[We've integrated functorch into PyTorch. As the final step of the 
integration, `functorch.zk` is deprecated as of PyTorch 
2.0 and will be deleted in a future version of PyTorch >= 2.3. 
Please use `z™` instead; see the PyTorch 2.0 release notes 
and/or the `torch.func` migration guide for more details 
https://pytorch.org/docs/main/func.migrating.htmlÚ
Ú )Úreplace)r   r   r   Úwarnings   &&& r   Úget_warningr      sR   € ð ‚Ø ˜uÐ%ˆð#Ø#& %ð (à�ið  <ð	=ð ÷ Ø—/‘/ $¨Ó+ˆØ€Nr   c               ó$   € V ^8„  d   QhRRRRRR/# )r
   r   r   r   r   r   ÚNoner   )r   s   "r   r   r   /   s!   € ÷ 8ñ 8˜ð 8 zð 8¸Tñ 8r   c                óZ   € \        WR R7      p\        P                  ! V\        ^R7       R# )T)r   )Ú
stacklevelN)r   ÚwarningsÚwarnÚFutureWarning)r   r   r   s   && r   Úwarn_deprecatedr$   /   s   € Ü˜#¸Ô>€GÜ‡M‚M�'œ=°Q×7r   c               ó(   € V ^8„  d   QhRRRRRRRR/# )	r
   Úfunctorch_apiúCallable[..., Any]Útorch_func_apizCallable[..., Any] | NoneÚnew_api_namer   r   r   r   )r   s   "r   r   r   4   s6   € ÷ Bñ BØ%ðBà-ðBð ðBð 
ñ	Br   c                ó  € V P                   pVf   \        \        V4      pVP                  f   R # \	        W24      pR\
        P                  ! VR4      ,           p\
        P                  ! VR4      pVP                  V,           V n        R # )Nz
.. warning::

z    )Ú__name__ÚgetattrÚ_implÚ__doc__r   ÚtextwrapÚindent)r&   r(   r)   Úapi_namer   Úwarning_notes   &&&   r   Ú
setup_docsr3   4   sr   € ð
 ×%Ñ%€HØÒÜ ¤¨Ó1ˆà×ÑÒ%Ùä˜(Ó1€GØ'¬(¯/ª/¸'À6Ó*JÕJ€LÜ—?’? <°Ó8€LØ*×2Ñ2°\ÕA€MÖr   ÚerrorÚ
chunk_sizec               ó0   € V ^8„  d   QhRRRRRRRRR	R
RR/# )r
   Úfuncr'   Úin_dimsr   Úout_dimsr   Ú
randomnessr   r5   ú
int | Noner   r   )r   s   "r   r   r   F   sL   € ÷ 	Qñ 	QØ
ð	Qàð	Qð ð	Qð ð		Qð ð	Qð ñ	Qr   c               óL   € \        R R4       \        P                  ! WW#VR7      # )Úvmapú
torch.vmap)r5   )r$   Úapisr=   )r7   r8   r9   r:   r5   s   &&&&$r   r=   r=   F   s!   € ô �F˜LÔ)Ü�9Š9�T HÀZÔPÐPr   c               ó(   € V ^8„  d   QhRRRRRRRR/# ©r
   r7   r'   Úargnumsr   Úhas_auxr   r   r   )r   s   "r   r   r   R   s,   € ÷ -ñ -Ø
ð-Ø'0ð-Ø?Cð-àñ-r   c                óF   € \        R 4       \        P                  ! WV4      # )Úgrad)r$   r?   rE   ©r7   rB   rC   s   &&&r   rE   rE   R   s   € ô �FÔÜ�9Š9�T GÓ,Ð,r   c               ó(   € V ^8„  d   QhRRRRRRRR/# rA   r   )r   s   "r   r   r   Y   s,   € ÷ 7ñ 7Ø
ð7Ø'0ð7Ø?Cð7àñ7r   c                óF   € \        R 4       \        P                  ! WV4      # )Úgrad_and_value)r$   r?   rI   rF   s   &&&r   rI   rI   Y   s    € ô Ð$Ô%Ü×Ò˜t¨gÓ6Ð6r   rC   c               ó(   € V ^8„  d   QhRRRRRRRR/# )r
   r7   r'   Úprimalsr   rC   r   r   r   )r   s   "r   r   r   `   s)   € ÷ 6ñ 6Ð ð 6¨Cð 6¸$ð 6È3ñ 6r   c               óL   € \        R 4       \        P                  ! V .VO5RV/ # )ÚvjprC   )r$   r-   rM   )r7   rC   rK   s   &$*r   rM   rM   `   s#   € Ü�EÔÜ�9Š9�TÐ5˜GÒ5¨WÑ5Ð5r   Ústrictc               ó0   € V ^8„  d   QhRRRRRRRRRRR	R/# )
r
   r7   r'   rK   r   ÚtangentsrN   r   rC   r   r   )r   s   "r   r   r   e   sL   € ÷ 	Nñ 	NØ
ð	Nàð	Nð ð	Nð
 ð	Nð ð	Nð 	ñ	Nr   c               óJ   € \        R 4       \        P                  ! WW#VR7      # )Újvp)rN   rC   )r$   r-   rR   )r7   rK   rP   rN   rC   s   &&&$$r   rR   rR   e   s   € ô �EÔÜ�9Š9�T HÀWÔMÐMr   Ú_preallocate_and_copyc               ó0   € V ^8„  d   QhRRRRRRRRR	RR
R/# )r
   r7   r'   rB   zint | tuple[int, ...]rC   r   r5   r;   rS   r   r   )r   s   "r   r   r   q   sD   € ÷ ñ Ø
ðà"ðð ð	ð
 ðð  ðð ñr   c               óN   € \        R 4       \        P                  ! V VVVVR7      # )Újacrev)rC   r5   rS   )r$   r-   rV   )r7   rB   rC   r5   rS   s   &&$$$r   rV   rV   q   s,   € ô �HÔÜ�<Š<ØØØØØ3ôð r   r:   c          
     ó,   € V ^8„  d   QhRRRRRRRRR	R/# )
r
   r7   r'   rB   r   rC   r   r:   r   r   r   )r   s   "r   r   r   ƒ   sA   € ÷ Gñ GØ
ðGàðGð ðGð
 ðGð ñGr   c               óH   € \        R 4       \        P                  ! WW#R7      # )Újacfwd)r:   )r$   r-   rY   )r7   rB   rC   r:   s   &&&$r   rY   rY   ƒ   s   € ô �HÔÜ�<Š<˜ wÔFÐFr   c               ó$   € V ^8„  d   QhRRRRRR/# )r
   r7   r'   rB   Úintr   r   )r   s   "r   r   r   Ž   s#   € ÷ 0ñ 0Ð$ð 0¨sð 0Ð;Mñ 0r   c                óF   € \        R 4       \        P                  ! WR7      # )Úhessian)rB   )r$   r-   r]   )r7   rB   s   &&r   r]   r]   Ž   s   € Ü�IÔÜ�=Š=˜Ô/Ð/r   ÚremoveÚ	mutationsc               ó$   € V ^8„  d   QhRRRRRR/# )r
   r7   r'   r^   r   r   r   )r   s   "r   r   r   “   s$   € ÷ 4ñ 4Ø
ð4Ø),ð4àñ4r   c               óF   € \        R 4       \        P                  ! WR7      # )Úfunctionalize)r^   )r$   r-   rb   )r7   r^   s   &$r   rb   rb   “   s   € ô �OÔ$Ü×Ò˜tÔ3Ð3r   c               ó$   € V ^8„  d   QhRRRRRR/# ©r
   Úmodelz	nn.ModuleÚdisable_autograd_trackingr   r   r   r   )r   s   "r   r   r   š   s'   € ÷ Fñ F˜9ð FÀð FÐRUñ Fr   c                óF   € \        R R4       \        P                  ! W4      # )Úmake_functionalútorch.func.functional_call)r$   Ú_nn_implrh   ©re   rf   s   &&r   rh   rh   š   s   € ÜÐ%Ð'CÔDÜ×#Ò# EÓEÐEr   c               ó$   € V ^8„  d   QhRRRRRR/# rd   r   )r   s   "r   r   r   Ÿ   s)   € ÷ Sñ SØðSØ15ðSàñSr   c                óF   € \        R R4       \        P                  ! W4      # )Úmake_functional_with_buffersri   )r$   rj   rn   rk   s   &&r   rn   rn   Ÿ   s!   € ô Ð2Ð4PÔQÜ×0Ò0°ÓRÐRr   c               ó    € V ^8„  d   QhRRRR/# )r
   Úmodelszlist[nn.Module]r   r   r   )r   s   "r   r   r   ¦   s   € ÷ 7ñ 7 ð 7¸3ñ 7r   c                óF   € \        R R4       \        P                  ! V 4      # )Úcombine_state_for_ensembleútorch.func.stack_module_state)r$   rj   rr   )rp   s   &r   rr   rr   ¦   s   € ÜÐ0Ð2QÔRÜ×.Ò.¨vÓ6Ð6r   r>   ri   rs   )NF)N)NN)é    rt   r4   )rt   F)rt   )F)'r.   Ú
__future__r   r/   r!   Útypingr   r   Útorch._functorch.apisÚ
_functorchr?   Ú!torch._functorch.eager_transformsÚeager_transformsr-   Ú torch._functorch.make_functionalrh   rj   Útorch.nnÚnnÚcollections.abcr   r   Útorch._functorch.vmapr   r   r   r$   r3   r=   rE   rI   rM   rR   rV   rY   r]   rb   rn   rr   r   r   r   Ú<module>r€      s–  ðñõ #ã Û ß %ç $Ð $ß 1Ð 1ß 3Ð 3Ý ÷ Ý(å;ß;÷÷$8÷
Bñ$	Qð "÷	Qð 	Q÷-÷7ð6À÷ 6ð
	Nð
 ð	Nð ÷	Nñð ð	ð
 "ðð #(÷ð ñ$Gð
 ÷Gð G÷0ð
4Ø/:÷4÷F÷
Sõ7ñ
 ˆ4�—‘˜LÔ )Ù 
ˆ4�—‘Ô Ù 
ˆ>˜4×.Ñ.Ô /Ù 
ˆ3„Ù 
ˆ3„Ù 
ˆ6Ô Ù 
ˆ6Ô Ù 
ˆ7Ô Ù 
ˆ=Ô Ù 
ˆ?˜H×4Ñ4Ð6RÔ SÙ 
Ø  (×":Ñ":Ð<Xôñ ØØ×'Ñ'Ø#ör   