Machine Learning in Elixir

Creating Neural Networks with Axon

Axon is a library for creating and training neural networks in Elixir and is the primary tool for deep learning in the Elixir ecosystem. For the remainder of this book, you’ll spend time diving deep into the Axon API and using Axon to create and train neural networks on complex problems. As you saw in the previous section, there’s a lot of boilerplate code associated with creating neural networks in Nx. Axon abstracts all of the boilerplate code and offers a simplified API for building and training neural networks.

As an introduction to deep learning in Elixir, you’ll train an algorithm that solves the problem of classifying handwritten digits, as described earlier in ​Understanding the Need for Deep Learning ...