Top 2 optax open source projects

The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.

✭ 229

Jupyter Notebook machine-learning tutorial deep-learning jupyter numpy flax haiku lax jax xla optax learn-jax

uvadlc notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2022/Spring 2022

✭ 901

Jupyter Notebook tutorial deep-learning tutorials pytorch flax jax pytorch-lightning optax uvadlc

1-2 of 2 optax projects