P/j`ddlmZmZmZmZmZmZmZddlm Z m Z ddl m Z m Z ddlmZddlmZy))vjpjvpjacrevjacfwdhessian functionalize linearize)gradgrad_and_value)functional_callstack_module_state)replace_all_batch_norm_modules_)vmapN)!torch._functorch.eager_transformsrrrrrrr torch._functorch.apisr r torch._functorch.functional_callr r 'torch._functorch.batch_norm_replacementrrV/root/repos/pocs/video-evals/.venv/lib/python3.12/site-packages/torch/func/__init__.pyrs"7PS&r