I've been spending my time recently updating the Axon training API to mirror the functionality of the amazing @pytorch_ignite library. You can check out the new API here: github.com/elixir-nx/axon… If you are remotely interested in Elixir or neural networks or both, I would be very happy if you helped me out by testing out the API with some fun examples and providing feedback, reporting issues, etc. :)