EnvLogger

Perceiver IO

Large-scale graph representation learning with very deep GNNs and self-supervision

Sparse reward tasks to learn behaviour priors

AlphaFold Protein Structure Database

AlphaFold

Melting Pot

AndroidEnv

Launchpad

Alchemy

Jraph

Specifying tensor shapes using Python type annotations

FRCL

3D simulated environments for fast grounded language learning

Self-supervised Multimodal Versatile Networks

ODE-GAN

Efficient and tight neural network verification in JAX

Jax_verify

DQN Zoo

Learning to Simulate Complex Physics with Graph Networks

Paired Associative Inference Task

Computational predictions of protein structures associated with COVID-19

Chex

RL Unplugged

Deep Ensembles