+
    –°³jÏ  ã                   ór   € ^ RI HtHtHt ^ RIHt ^ RIHtHtH	t	H
t
HtHtHtHt ^ RIHt ^ RIHtHt . ROtR# )é    )ÚgradÚgrad_and_valueÚvmap)Úreplace_all_batch_norm_modules_)Údebug_unwrapÚfunctionalizeÚhessianÚjacfwdÚjacrevÚjvpÚ	linearizeÚvjp)Ú	rearrange)Úfunctional_callÚstack_module_stateN)r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   )Útorch._functorch.apisr   r   r   Ú'torch._functorch.batch_norm_replacementr   Ú!torch._functorch.eager_transformsr   r   r	   r
   r   r   r   r   Útorch._functorch.einopsr   Ú torch._functorch.functional_callr   r   Ú__all__© ó    ÚT/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/func/__init__.pyÚ<module>r      s,   ðß <Ñ <Ý S÷	÷ 	ó 	õ .ß Pò‚r   