GitHub - elixir-nx/axon: Nx-powered Neural Networks.
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...
Source: github.com