learn-jax

There are 0 repository under learn-jax topic.

• get-started-with-JAX

  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.

  deep-learningflaxhaikujaxjupyterlaxlearn-jaxmachine-learningnumpyoptaxpythontutorialxla
  Language:Jupyter Notebook 586