learn-jax

Here is 1 public repository matching this topic...

gordicaleksa / get-started-with-JAX

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.

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

Updated Nov 29, 2023
Jupyter Notebook

Improve this page

Add a description, image, and links to the learn-jax topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the learn-jax topic, visit your repo's landing page and select "manage topics."

Learn more

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

learn-jax

Here is 1 public repository matching this topic...

gordicaleksa / get-started-with-JAX

Improve this page

Add this topic to your repo