← all repositories

elixir-nx/axon

A neural network library for Elixir providing APIs for model creation, training, and inference using the Nx numerical computing framework.

1.7k stars Elixir ML Frameworks
axon
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

Axon is a deep learning framework for Elixir that provides functional and high-level APIs for building and training neural networks. It offers components for activations, layers, losses, metrics, and initialization, all implemented as numerical definitions compatible with Nx compilers and backends. The framework is inspired by PyTorch Ignite and uses Polaris for its optimization API.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.