Scenic: A Jax Library for Computer Vision Research and Beyond
updated at May 10, 2024, 1:34 p.m.
Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.
updated at May 10, 2024, 10:47 a.m.
A library for programmatically generating equivariant layers through constraint solving
updated at May 10, 2024, 10:02 a.m.
Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
updated at May 10, 2024, 3:07 a.m.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
updated at May 9, 2024, 5:07 p.m.
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
updated at May 9, 2024, 12:33 p.m.
Probabilistic Programming and Nested sampling in JAX
updated at May 9, 2024, 11:54 a.m.
A Python package of computer vision models for the Equinox ecosystem.
updated at May 8, 2024, 8:14 p.m.
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
updated at May 8, 2024, 6:50 p.m.
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
updated at May 8, 2024, 3:34 p.m.