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. :)
Also, I tried to be as thorough as possible with explanations in the documentation. If anything is unclear, please let me know!
Finally, you can read more about the inspiration and concepts taken from PyTorch Ignite here: pytorch.org/ignite/index.h…