ó
    Eñi�  ã                   ód   • S SK JrJrJr  S SKJr  S SKJrJrJ	r	J
r
JrJrJrJr  S SKJrJr  / SQrg)é    )ÚgradÚgrad_and_valueÚvmap)Úreplace_all_batch_norm_modules_)Údebug_unwrapÚfunctionalizeÚhessianÚjacfwdÚjacrevÚjvpÚ	linearizeÚvjp)Úfunctional_callÚstack_module_state)r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   N)Útorch._functorch.apisr   r   r   Ú'torch._functorch.batch_norm_replacementr   Ú!torch._functorch.eager_transformsr   r   r	   r
   r   r   r   r   Ú torch._functorch.functional_callr   r   Ú__all__© ó    ÚP/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/func/__init__.pyÚ<module>r      s*   ðß <Ñ <Ý S÷	÷ 	ó 	÷ Qò�r   