xref: /aosp_15_r20/external/executorch/extension/llm/modules/README.md (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1## Export-friendly Modules
2
3Modules in this directory:
4* Extend `torch.nn.Module`.
5* Are guaranteed to work out of the box with `torch.export.export()`.
6* Should work out of the box with `torch.aot_compile()`.
7* Should be able to workt with ExecuTorch.
8
9All modules should be covered by unit tests to make sure they are:
101. Give the output as the reference eager model in PyTorch or TorrchTune
112. Export-friendly
12
13Additionally, we aim to make these modules:
143. AOTI-friendly
154. ExecuTorch-friendly
16
17These modules are subject to change (may upstream to TorchTune) so proceed with caution.
18