  toggle dark mode toggle dark mode

Tag: Jax

A Fully connected neural network in JAX

posted on 2020-12-25T14:48:19Z
tags: ml (7)machine learning python (6) cv (3)computer vision jax (1)

JAX 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…