o Iñ hã@s<ddlmZddlmZddlmZddlmZmZm Z dS)é)Ú functionalize)Ú chunk_vmap)Úreplace_all_batch_norm_modules_)ÚhessianÚjacfwdÚjvpN) Ú functorchrÚtorch._functorch.apisrÚ'torch._functorch.batch_norm_replacementrÚ!torch._functorch.eager_transformsrrr©r r úS/var/www/vscode/kcb/lib/python3.10/site-packages/functorch/experimental/__init__.pyÚs