Menu
Back to Open Source

🐙 GitHub Detail

G

gordicaleksa/get-started-with-JAX

By gordicaleksa

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.

GitHub Jupyter Notebook MIT License Updated 30 May 2026

Live Snapshot

Stars

784

🍴

Forks

119

📄

License

MIT License

🧩

Type

Jupyter Notebook

📘

About this open-source project

Live information fetched from GitHub.

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.

🌿

Default Branch

main

🐞

Open Issues

1

👀

Watchers

784