penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

mujoco_menagerie

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.

bsuite

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent

educational

jraph

A Graph Neural Network Library in Jax

tapnet

Tracking Any Point (TAP)

rc-data

Question answering dataset featured in "Teaching Machines to Read and Comprehend

rlax

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control

android_env

RL research on Android devices.

scalable_agent

A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.

neural-processes

This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).