get-started-with-JAX

get-started-with-JAX

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.

755 Stars
114 Forks
755 Watchers
Jupyter Notebook Language
mit License
Cost to Build
$7.8K
Market Value
$28.4K

Growth over time

5 data points  ·  2022-03-01 → 2025-09-01
Stars Forks Watchers
💬

How do you feel about this project?

Ask AI about get-started-with-JAX

Question copied to clipboard

What is the gordicaleksa/get-started-with-JAX GitHub project? Description: "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.". Written in Jupyter Notebook. Explain what it does, its main use cases, key features, and who would benefit from using it.

Question is copied to clipboard — paste it after the AI opens.

How to clone get-started-with-JAX

Clone via HTTPS

git clone https://github.com/gordicaleksa/get-started-with-JAX.git

Clone via SSH

[email protected]:gordicaleksa/get-started-with-JAX.git

Download ZIP

Download master.zip

Found an issue?

Report bugs or request features on the get-started-with-JAX issue tracker:

Open GitHub Issues