o êÜÓhã@s<ddlmZddlmZddlmZmZmZddlm Z dS)é)Ú chunk_vmap)Úreplace_all_batch_norm_modules_)ÚhessianÚjacfwdÚjvp)Ú functionalizeN) Útorch._functorch.apisrÚ'torch._functorch.batch_norm_replacementrÚ!torch._functorch.eager_transformsrrrÚ functorchr©r r úU/var/www/html/ai/venv/lib/python3.10/site-packages/functorch/experimental/__init__.pyÚs