bi<ddlmZddlmZddlmZddlmZmZm Z y)) functionalize) chunk_vmap)replace_all_batch_norm_modules_)hessianjacfwdjvpN) functorchrtorch._functorch.apisr'torch._functorch.batch_norm_replacementr!torch._functorch.eager_transformsrrrZ/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/functorch/experimental/__init__.pyrs#,SBBr