
    ^j                     p    d dl mZmZmZ d dlmZ d dlmZmZm	Z	m
Z
mZmZmZmZ d dlmZ d dlmZmZ g dZy)    )gradgrad_and_valuevmap)replace_all_batch_norm_modules_)debug_unwrapfunctionalizehessianjacfwdjacrevjvp	linearizevjp)	rearrange)functional_callstack_module_state)r   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.einopsr    torch._functorch.functional_callr   r   __all__     ^/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/func/__init__.py<module>r      s,    < < S	 	 	 . Pr   