Axon
A neural network library for Elixir built on Nx, providing functional, model creation, and training APIs for deep learning.
What is Axon?
Axon is a neural network library for Elixir that provides tools for building, training, and deploying deep learning models. It solves the problem of bringing modern deep learning capabilities to the Elixir ecosystem by offering functional, model creation, and training APIs powered by Nx for efficient numerical computation.
Elixir developers and data scientists who want to implement deep learning models within the Elixir ecosystem, leveraging its concurrency and fault-tolerance features for machine learning workloads.
Developers choose Axon because it integrates seamlessly with Elixir and Nx, offering a clean API inspired by PyTorch, full control over model components, and the ability to compile models using various Nx backends for performance optimization.
Overview
Nx-powered Neural Networks
Use Cases
Best For
- Building neural networks in Elixir with high-level abstractions
- Training deep learning models using Elixir's concurrency model
Related Projects
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
