Implementation of Diffusion Transformer (DiT) in JAX

Multi-Joint dynamics with Contact. A general purpose physics simulator.

The official Meta Llama 3 GitHub site

Synthesizing Moving People with 3D Control

Code for "Interactive Task Planning with Language Models"

Canonical Factors for Hybrid Neural Fields @ ICCV 2023

Building blocks for productive research

MTM Masked Trajectory Models for Prediction, Representation, and Control.

AI-powered code assistant for Vim. OpenAI and ChatGPT plugin for Vim and Neovim.

[CoRL '23] Dexterous piano playing with deep reinforcement learning.

A tiny reinforcement learning codebase for continuous control, built on top of JAX.

Real-time behaviour synthesis with MuJoCo, using Predictive Control

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

Building Open-Ended Embodied Agents with Internet-Scale Knowledge

A CLI for processing composite Wavefront OBJ files for use in MuJoCo.

Instant neural graphics primitives: lightning fast NeRF and more

Zero-effort CLI interfaces & config objects, from types

Neovim plugin for GitHub Copilot

GPT implementation in Flax

Async isort plugin for Vim + Neovim

tensorflow code for differentiable filtering

Pytrees + dataclasses ❤️

Generic EKF, with support for non-Euclidean manifolds

Factor graphs and nonlinear least squares for JAX

Declarative, type-safe command line argument parsers from dataclasses and attrs classes