elixir-nx/axon

Health v2 · maintenance only

Adoption not part of the score

1688 stars · 129 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

Axon is a neural network library for Elixir built on top of Nx numerical definitions. It provides a functional API, a high-level model creation API, and a PyTorch Ignite-inspired training API, all composable and compilable via Nx backends.

Use cases

• build and train neural networks in Elixir
• define deep learning models with a high-level API
• train models with custom loss functions and optimizers
• export models to ONNX or TensorFlow Lite
• compose custom numerical definitions with standard layers
• accelerate training with Nx compilers and backends

When to avoid

• you need a mature ecosystem of pretrained models like PyTorch or TensorFlow
• you require GPU tooling and community support beyond the Nx ecosystem
• you want a framework with extensive production deployment tooling

Facets

library · maturity active

deep-learning machine-learning llm-training deep-learning machine-learning artificial-intelligence elixir jvm-scripting neural-networks nx elixir optimizers jit-compilation onnx-export

Member repositories

For agents

markdown · JSON · MCP: product_card(name="elixir-nx/axon")