turbocharge_ml_jax_tpu

ML with JAX and TPUs course

Materials live here.

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
_book		_book
docs		docs
images		images
.gitignore		.gitignore
.nojekyll		.nojekyll
JAX_foundations.ipynb		JAX_foundations.ipynb
README.md		README.md
Resnet.ipynb		Resnet.ipynb
TPUs.ipynb		TPUs.ipynb
_quarto.yml		_quarto.yml
cover.png		cover.png
diffusion.ipynb		diffusion.ipynb
exercise1_jax_mnist.ipynb		exercise1_jax_mnist.ipynb
exercise1_jax_mnist_solution.ipynb		exercise1_jax_mnist_solution.ipynb
exercise2_linear_reg_flax.ipynb		exercise2_linear_reg_flax.ipynb
exercise2_solution.ipynb		exercise2_solution.ipynb
flax_foundations.ipynb		flax_foundations.ipynb
index.qmd		index.qmd
jax_linear_regression.ipynb		jax_linear_regression.ipynb
jax_xla.png		jax_xla.png
references.bib		references.bib
references.qmd		references.qmd
stable_diffusion.ipynb		stable_diffusion.ipynb
summary.qmd		summary.qmd
updated_cover.png		updated_cover.png
vit_jax.ipynb		vit_jax.ipynb

rastringer/turbocharge_ml_jax_tpu