Writing an MCMC sampler in JAX by Jeremie Coullon
Overview
Tutorial on the different ways to write an MCMC sampler in JAX along with speed benchmarks
Related Projects
Get started with JAX by Aleksa Gordić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.
Extending JAX with custom C++ and CUDA code by Dan Foreman-MackeyExtending JAX with custom C++ and CUDA code
Tutorial: image classification with JAX and Flax Linen by 8bitmp3How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
Ensemble networks by Mat KelceyEnsemble nets are a method of representing an ensemble of models as one single logical model
Found a gem we're missing?
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.