Repository · GitHub

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.

View on GitHubReport an issue

Owner	gordicaleksa
License (as declared)	MIT
Stars	790
Forks	121
Main language	Jupyter Notebook
Created	28 October 2021
Last updated at source	29 November 2023

jupyter notebookdeep-learningflaxhaikujaxjupyterlaxlearn-jaxmachine-learningnumpyoptaxpythontutorialxla

Embed this entry

<iframe src="https://silytics.si/embed/item/822ca689-1e04-4594-a590-fb3c507fc955" width="420" height="190" style="border:0" loading="lazy" title="Silytics entry"></iframe>

Free, no tracking. More about feeds and embeds.

Stars, last 5 days

790 now. Counts as reported by the source, snapshotted daily by Silytics.

More from gordicaleksa

Silytics shows public metadata from GitHub and links to the original. Information may be incomplete or out of date. Last refreshed 20 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.