Google DeepMind
Google DeepMind
wasserstein_fairness
An implementation of Wasserstein Fair Classification, a conference paper submitted to UAI 2019.
long-form-factuality
Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".
optax
Optax is a gradient processing and optimization library for JAX.
penzai
A JAX research toolkit for building, editing, and visualizing neural networks.
nuclease_design
ML-guided enzyme engineering
mujoco_warp
GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.
videoprism
Official repository for "VideoPrism: A Foundational Visual Encoder for Video Understanding" (ICML 2024)