GitHub - Wesley-Jakob-Gilbert/JAX-Learning: A series of Jupyter Notebooks where I went through the JAX documentation tutorials, gave my inputs and generally played around a bit.

#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.

Name		Name	Last commit message	Last commit date
.gitignore		.gitignore
How To Scale Your Model PSet - Chapter 1 All About Rooflines.pdf		How To Scale Your Model PSet - Chapter 1 All About Rooflines.pdf
JAX - the sharp bits.ipynb		JAX - the sharp bits.ipynb
JAX Automatic Differentiation.ipynb		JAX Automatic Differentiation.ipynb
JAX Pytrees.ipynb		JAX Pytrees.ipynb
Jax 101 - JIT.ipynb		Jax 101 - JIT.ipynb
Jax_Quickstart.ipynb		Jax_Quickstart.ipynb
README.md		README.md

Navigation Menu

Repository files navigation

About

Releases

Packages

Contributors

Languages

Latest commit

History

Folders and files

Repository files navigation

About

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages