GitHub - elixir-nx/axon: Nx-powered Neural Networks

Nx-powered Neural Networks for Elixir. 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. Training API – An API for quickly training models, inspired by PyTorch Ignite. Axon provides abstractions that ena