Name Date Size #Lines LOC

..--

test/H25-Apr-2025-475373

README.mdH A D25-Apr-2025588 1813

__init__.pyH A D25-Apr-2025751 2517

_position_embeddings.pyH A D25-Apr-202529.2 KiB728541

attention.pyH A D25-Apr-202517 KiB414325

kv_cache.pyH A D25-Apr-20255.7 KiB149117

README.md

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