Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
created at May 24, 2022, 10:26 p.m.
đšī¸ A diverse suite of scalable reinforcement learning environments in JAX
created at Aug. 11, 2022, 7:34 a.m.
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.
created at Oct. 28, 2021, 8 p.m.
Really Fast End-to-End Jax RL Implementations
created at Feb. 25, 2023, 3:38 p.m.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
created at Jan. 16, 2021, 1:26 a.m.
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
created at Dec. 9, 2020, 8:12 a.m.