U
    Ò9%e‘  ã                   @   s`   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 d dlmZ d dl	mZ dS )é    )Úgrad_and_valueÚvjpÚjvpÚjacrevÚjacfwdÚhessianÚfunctionalizeÚ	linearize)Úgrad)Úfunctional_callÚstack_module_state)Úreplace_all_batch_norm_modules_)ÚvmapN)Z!torch._functorch.eager_transformsr   r   r   r   r   r   r   r	   Ztorch._functorch.apisr
   Z torch._functorch.functional_callr   r   Z'torch._functorch.batch_norm_replacementr   r   © r   r   úR/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/torch/func/__init__.pyÚ<module>   s   (
