Metadata-Version: 2.4
Name: axelera-axaten
Version: 1.2.2
Summary: Axelera ATen functional operators
License-Expression: LicenseRef-Proprietary
Requires-Python: >=3.10
Description-Content-Type: text/markdown
License-File: LICENSE.txt
Requires-Dist: numpy
Requires-Dist: pydantic
Requires-Dist: torch<2.10.0,>=2.8.0
Requires-Dist: pandas
Provides-Extra: test
Requires-Dist: pytest>=7.0; extra == "test"
Requires-Dist: transformers<5.8,>=4.57; extra == "test"
Dynamic: license-file

# axelera-axaten

Axelera ATen functional operators: PyTorch reference implementations of the
operators the Axelera compiler supports (quantize/dequantize, layer and RMS
normalisation, attention, rotary embeddings, KV cache) and the piecewise-linear
approximation tables used for non-linear functions.

## License

This software is governed by the Axelera AI End User License Agreement, see
[LICENSE.txt](LICENSE.txt) and <https://www.axelera.ai/eula>.
