craffel / jax-tutorial

A tutorial on JAX (https://github.com/google/jax/)

JAX Tutorial

This repository contains a Jupyter notebook tutorial to JAX. You can view a rendered version of the notebook here.