a h@s<ddlmZddlmZddlmZddlmZmZm Z dS)) functionalize) chunk_vmap)replace_all_batch_norm_modules_)hessianjacfwdjvpN) Z functorchrZtorch._functorch.apisrZ'torch._functorch.batch_norm_replacementrZ!torch._functorch.eager_transformsrrrrrM/var/www/auris/lib/python3.9/site-packages/functorch/experimental/__init__.pys