Shypd.aiThe largest open index of AI toolsAxon is an AI Frameworks & Infra tool that provides Nx-powered neural networks for Elixir. It offers a functional API for numerical definitions, a model creation API, and a training API inspired by PyTorch Ignite.
Trending
About
Axon is a comprehensive library designed for building and training neural networks within the Elixir programming language, leveraging the power of Nx. It provides a modular architecture with distinct functional, model creation, and training APIs, allowing developers full control over each aspect of neural network development. The functional API includes modules for activations, initializers, layers, losses, and metrics, all implemented as numerical definitions that can be accelerated by any Nx compiler or backend. The model creation API simplifies the definition and serialization of neural network models, while the training API, inspired by PyTorch Ignite, offers conveniences for implementing and instrumenting training loops. Axon integrates with Polaris for optimization and supports compilation to formats like TensorFlow Lite and ONNX, making it a robust solution for deep learning in Elixir.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
