| functorch.compile (experimental) |
| ================================ |
| |
| AOT Autograd is an experimental feature that allows ahead of time capture of |
| forward and backward graphs, and allows easy integration with compilers. This |
| creates an easy to hack Python-based development environment to speedup training |
| of PyTorch models. AOT Autograd currently lives inside ``functorch.compile`` |
| namespace. |
| |
| .. warning:: |
| AOT Autograd is experimental and the APIs are likely to change. We are looking |
| for feedback. If you are interested in using AOT Autograd and need help or have |
| suggestions, please feel free to open an issue. We will be happy to help. |
| |
| .. currentmodule:: functorch.compile |
| |
| Compilation APIs (experimental) |
| ------------------------------- |
| .. autosummary:: |
| :toctree: generated |
| :nosignatures: |
| |
| aot_function |
| aot_module |
| memory_efficient_fusion |
| |
| Partitioners (experimental) |
| --------------------------- |
| .. autosummary:: |
| :toctree: generated |
| :nosignatures: |
| |
| default_partition |
| min_cut_rematerialization_partition |
| |
| Compilers (experimental) |
| ------------------------ |
| .. autosummary:: |
| :toctree: generated |
| :nosignatures: |
| |
| nop |
| ts_compile |
