pytorch/torch/_functorch
2025-01-23 23:34:13 +00:00
..
_activation_checkpointing PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
_aot_autograd Revert "[compiled autograd] Always proxy autograd.Function nodes; handle AOT backwards (#143405)" 2025-01-23 23:34:13 +00:00
__init__.py
aot_autograd.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
apis.py [BE] typing for decorators (#144161) 2025-01-04 16:40:09 +00:00
autograd_function.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
batch_norm_replacement.py remove allow-untyped-defs for torch/_functorch/batch_norm_replacement.py (#143438) 2024-12-18 09:01:06 +00:00
benchmark_utils.py
compile_utils.py
compilers.py Remove unused Python variables in torch/[_-a]* (#133492) 2024-12-12 17:39:14 +00:00
config.py [reland][export] don't decompose custom triton op when exporting (#144284) 2025-01-11 01:34:35 +00:00
deprecated.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
eager_transforms.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
functional_call.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
fx_minifier.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
make_functional.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
partitioners.py [BE]: Simplify set add with set update (#145152) 2025-01-23 20:18:13 +00:00
pyfunctorch.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
python_key.py
pytree_hacks.py
top_operators_github_usage.py
utils.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00
vmap.py PEP585 update - torch/_functorch (#145139) 2025-01-19 07:06:10 +00:00