Installation

Run the following to install and setup the environment

conda create -n jax_tutorial -y
conda activate jax_tutorial
pip install -r requirements.txt
python -m ipykernel install --user --name jax_tutorial --display-name "Python (jax_tutorial)"

Running

The tutorials can be run as Jupyter notebooks

jupyter notebook