Axon

https://github.com/elixir-nx/axon/raw/main/axon.png

Axon consists of the following components:

Functional API – A low-level API of numerical definitions (defn) of which all other APIs build on.

Model Creation API – A high-level model creation API which manages model initialization and application.

Optimization API – An API for creating and using first-order optimization techniques.

Training API – An API for quickly training models.

Axon provides abstractions that enable easy integration while maintaining a level of separation between each component. You should be able to use any of the APIs without dependencies on others. By decoupling the APIs, Axon gives you full control over each aspect of creating and training a neural network.