Skip to main content

An ahead-of-time nn compiler for Ambiq SoCs.

Project description

heliaAOT

Bring blazing-fast, ultra-compact neural inference to Ambiq’s ultra-low-power SoCs—without a runtime.

CI Docs Build Wheel Release

heliaAOT is an ahead-of-time (AOT) compiler that converts front-end models (e.g., TFLite) into stand-alone C inference modules optimized for Ambiq microcontrollers. By shifting codegen to build-time and eliminating the runtime interpreter, heliaAOT delivers compact, readable C with zero arena-guesswork and no dead code.


✨ Why heliaAOT?

  • Smaller binaries – Generate only the code you need. No interpreter, no dead kernels.
  • Deterministic memory – Context-level memory planning; stop guessing arena sizes.
  • Readable output – Layer-mirrored C files for fast bring-up and debugging.
  • Ambiq-tuned – Optimized for Cortex-M (M4/M55/Helium) on Ambiq SoCs.
  • Production-ready – Drop into Zephyr/neuralSPOT or your existing build system.

🚀 Quick Start

Option A: Install via pipx

pipx install --python python3.12 helia-aot
helia-aot --help

We recommend pipx because it installs Python apps in isolated, self-contained environments and puts the CLI on your PATH.

Option B: Install via uv

uv tool install --python python3.12 helia-aot
helia-aot --help

Option C: Install via pip

python -m venv .venv
source .venv/bin/activate  # Windows: .venv\Scripts\activate
python -m pip install helia-aot
helia-aot --help

Option D: Install from source

git clone https://github.com/AmbiqAI/helia-aot.git
cd helia-aot
uv sync
uv run helia-aot --help

🧭 First Conversion

helia-aot convert \
  --model.path ./models/ad01_int8.tflite \
  --module.path ./out \
  --module.name ad01_aot

What you get:

out/ad01_aot/
├── LICENSE               # License for the generated module
├── README.md             # Module usage notes
├── includes-api/         # Public C headers
├── src/                  # Operator implementations
└── module.mk

Drop the folder into your firmware tree and build like any other C module.


🛠️ Current CLI

heliaAOT currently exposes a single primary command:

  • convert – Convert a TFLite model into a C inference module.

    helia-aot convert --help
    

    We follow “progressive disclosure of complexity.” The CLI starts simple; advanced flags and expert workflows are documented in the site’s How-to and Reference sections.


📚 Documentation

  • Getting Started: the fastest way from model → C module
  • How-To: step-by-step recipes for common tasks
  • Reference: full list of commands/flags
  • Guides: end-to-end projects and best practices
  • Benchmarks: performance and size comparisons
  • API: for power users who integrate deeper

The docs are written for CLI users first; you don’t need to know (or care) that the tool is implemented in Python.


🔌 Supported Inputs & Targets

Inputs

  • LiteRT/TFLite (int8/int16)

Targets

  • Ambiq SoCs (Cortex-M4, M55/Helium)
  • Integration paths: Zephyr module, neuralSPOT, or make/CMake projects

🧩 What the Generated C Looks Like

  • A compact operator library: only kernels required by your model.
  • A structured init() / run() style API with clear I/O.
  • Compile-time tensor shapes and a fixed memory plan.
  • Optional operator callbacks for instrumentation/telemetry.

🧪 Testing & Quality

  • Continuous Integration runs linting and unit tests
  • Deterministic builds and reproducible outputs where possible
  • Example apps and benchmarks in the docs

⚙️ Development (optional)

If you do want to contribute or run from source:

# Using uv (fast Python package manager)
uv sync
uv run helia-aot --help

# Or with pip + virtualenv
python -m venv .venv && source .venv/bin/activate
pip install -e ".[dev]"
helia-aot --help

🗺️ Roadmap

The project roadmap and features can be found on the GitHub Projects page.

We prioritize features that keep binaries tiny, predictable, and fast.


💬 Support & Feedback


📄 License

heliaAOT generates C modules that include a license header restricting use to Ambiq hardware. See the generated module’s LICENSE.txt for terms and the repository’s LICENSE for tooling.


Ready to dive in? Head over to the Getting Started guide and generate your first module in minutes.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

helia_aot-0.18.0.tar.gz (288.1 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

helia_aot-0.18.0-py3-none-any.whl (437.9 kB view details)

Uploaded Python 3

File details

Details for the file helia_aot-0.18.0.tar.gz.

File metadata

  • Download URL: helia_aot-0.18.0.tar.gz
  • Upload date:
  • Size: 288.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for helia_aot-0.18.0.tar.gz
Algorithm Hash digest
SHA256 fc6218f8cd6abf23de55e71880cbe5306bda6b86042e99042f8ec45a7cc74aed
MD5 0b0847f7b276faf689491773747c2c05
BLAKE2b-256 bb651e8ccffe1b43d26289282450ebc377bbf0dfa16957d746fa89923d029b13

See more details on using hashes here.

Provenance

The following attestation bundles were made for helia_aot-0.18.0.tar.gz:

Publisher: release.yml on AmbiqAI/helia-aot

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file helia_aot-0.18.0-py3-none-any.whl.

File metadata

  • Download URL: helia_aot-0.18.0-py3-none-any.whl
  • Upload date:
  • Size: 437.9 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.13

File hashes

Hashes for helia_aot-0.18.0-py3-none-any.whl
Algorithm Hash digest
SHA256 6b8483ff5ac86baf29148c56a5f37cffc7a070e8e74fe685a52f3e93785b829d
MD5 7e83e9b09d10c0036fd0a0b0fb72be47
BLAKE2b-256 3e26904a27cd426a8539617a365ca499e311f12ff139afb5dba564b4746616e1

See more details on using hashes here.

Provenance

The following attestation bundles were made for helia_aot-0.18.0-py3-none-any.whl:

Publisher: release.yml on AmbiqAI/helia-aot

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page