U Kºcíã@s<ddlmZddlmZmZmZddlmZddlm Z dS)é)Úreplace_all_batch_norm_modules_é)ÚjvpÚjacfwdÚhessian)Ú chunk_vmapé)Ú functionalizeN) Zbatch_norm_replacementrZ_src.eager_transformsrrrZ _src.vmaprZ functorchr ©r r úC/tmp/pip-unpacked-wheel-gikjz4vx/functorch/experimental/__init__.pyÚs