axon
axon copied to clipboard
elixir-nx
Nx-powered Neural Networks
Currently we abstract a lot of what goes on within RNNs away to a single layer call with some parameters. Our API is similar to Keras. I am considering making...
See also https://pytorch.org/tutorials/intermediate/parametrizations.html and https://keras.io/api/layers/constraints/
As I go through a major rewrite of the Axon compiler in #260, I've realized it will be very easy to make breaking changes if we do not include versions...
Hey, resuming from an existing loop state does not seem to work properly right now. For this, I am working in the horses_or_humans example. I am returning the entire state...
Willing to accept examples on different datasets and models to demonstrate different parts of the Axon API and to demonstrate Axon's viability in the ecosystem. The TensorFlow guides are a...
Hi everyone, Thanks for this excellent library! This PR aims to add an example for training the KuzushijiMNIST dataset using Axon(#47). The file is almost an exact replica of `mnist.exs`,...
As one example `validate` forwards metrics to the outer loop state and thus after it runs the loop step needs to be recompiled. This is possible in other handlers as...
There are some stability issues during training that I am investigating. Hopefully not due to grad :D Performance wise this isn't bad!