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.
AI summary
JAX tutorial series
A repository providing tutorials and resources to learn JAX, a popular alternative to PyTorch and TensorFlow for machine learning.
- stars
- 670
- forks
- 97
- watching
- 9
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
GP model library
Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation
RL library
Provides JAX implementations of various reinforcement learning algorithms with continuous action spaces.
Deep Learning Models
Provides a collection of deep learning models and utilities in JAX/Flax for research purposes.
JAX helper library
A set of utilities for writing reliable JAX code
Model repository
Provides pre-trained deep learning models for the Jax/Flax ecosystem.
JAX wrapper
This project provides infrastructure to interface custom C++ and CUDA code with the JAX library for scientific computing
Java tutorial repository
A repository containing various Java tutorials on topics such as Elasticsearch, gRPC, security, and serialization.
Kernel library
A set of composable kernels for scikit-learn implemented in JAX to accelerate computation and gradient calculation.
JAX progress bar library
Library that enhances JAX with support for dynamic progress bars in scans and loops
Graph library
A lightweight library for working with graph neural networks in jax.
Deep Learning Tutorials
A collection of tutorials and resources on implementing deep learning models using Python libraries such as Keras and Lasagne.
Hopsworks tutorials
A collection of tutorials and notebooks providing hands-on experience with the Hopsworks Platform for machine learning development and data analysis.
JAX toolbox
Provides optimized tools and infrastructure for JAX development on NVIDIA GPUs
Declarative GUI builder
A Haskell package providing a declarative programming framework for building GTK+ applications.
SNN library
A JAX-based library for training and utilizing spiking neural networks
Featured in 1 awesome list
Each link jumps to the spot where the list mentions get-started-with-JAX.
