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

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.