__init__.py
__pycache__/
_activation_checkpointing/
_activation_offloading/
_aot_autograd/
aot_autograd.py
apis.py
autograd_function.py
batch_norm_replacement.py
benchmark_utils.py
compile_utils.py
compilers.py
config.py
deprecated.py
eager_transforms.py
functional_call.py
fx_minifier.py
make_functional.py
partitioners.py
predispatch.py
pyfunctorch.py
python_key.py
pytree_hacks.py
top_operators_github_usage.py
utils.py
vmap.py