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.
JSON API: http://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/gordicaleksa%2Fget-started-with-JAX
PURL: pkg:github/gordicaleksa/get-started-with-JAX
Stars: 748
Forks: 113
Open issues: 1
License: mit
Language: Jupyter Notebook
Size: 1.78 MB
Dependencies parsed at: Pending
Created at: almost 4 years ago
Updated at: 11 days ago
Pushed at: almost 2 years ago
Last synced at: about 2 hours ago
Topics: deep-learning, flax, haiku, jax, jupyter, lax, learn-jax, machine-learning, numpy, optax, python, tutorial, xla