#Proof of Learning JAX#

This repository documents my JAX code package learning process, simple as that. I went through each of the JAX documentation tutorials, pasted code into Jupyter notebooks for each section, added some input, played around a little for better understanding, and then pushed to this repository.