o i@s<ddlmZddlmZddlmZddlmZmZm Z dS)) functionalize) chunk_vmap)replace_all_batch_norm_modules_)hessianjacfwdjvpN) functorchrtorch._functorch.apisr'torch._functorch.batch_norm_replacementr!torch._functorch.eager_transformsrrrr r \/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/functorch/experimental/__init__.pys