Tag: Jax
A Fully connected neural network in JAX
posted on 2020-12-25T14:48:19ZJAX is the hot new ML-tool on the block. I’m currently trying to get acquinted with it. The first thing I usually do to get to know a tool like this is to make a simple neural network to solve the MNIST digit recognition task. So here we go…
