deeplab2

DeepLab2 is a TensorFlow library for deep labeling, aiming to provide a unified and state-of-the-art TensorFlow codebase for dense pixel labeling tasks.

augmix

AugMix: A Simple Data Processing Method to Improve Robustness and Uncertainty

magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer

batch-ppo

Efficient Batched Reinforcement Learning in TensorFlow

pix2seq

Pix2Seq codebase: multi-tasks with generative modeling (autoregressive and diffusion)

seed_rl

SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's architecture.

rliable

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.

meta-dataset

A dataset of datasets for learning to learn from few examples

noisystudent

Code for Noisy Student Training. https://arxiv.org/abs/1911.04252

recsim

A Configurable Recommender Systems Simulation Platform

jax3d

long-range-arena

Long Range Arena for Benchmarking Efficient Transformers