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.
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.
SPU (Secure Processing Unit) aims to be a provable, measurable secure computation device, which provides computation ability while keeping your private data protected.
created at July 2, 2022, 9:05 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.
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
created at Dec. 9, 2020, 8:12 a.m.