SAPIEN Manipulation Skill Framework, a open source GPU parallelized robotics simulator and benchmark, led by Hillbot, Inc.

An open-source library for GPU-accelerated robot learning and sim-to-real transfer.

Official implementation of NerualSVG

(Crafter + NetHack) in JAX. ICML 2024 Spotlight.

A generative world for general-purpose robotics & embodied AI learning.

Goal-Conditioned Reinforcement Learning with JAX

Interpolation and function approximation with JAX

The math library of Lean 4

Tool for data extraction and interacting with Lean programmatically.

Scientific computing in Lean 4

Lean 4 programming language and theorem prover

"A modular and clean research framework for Generative AI, based on popular open-source repositories like Transformers, TRL, Diffusions, and CleanRL."

Imitation learning benchmark focusing on complex locomotion tasks using MuJoCo.

Clean single-file implementation of offline RL algorithms in JAX

🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL

Lark is a parsing toolkit for Python, built with a focus on ergonomics, performance and modularity.

Code repo for Gradient Temporal-Difference Learning with Regularized Corrections paper.

Successive Convexification General Framework

Universal Tensor Operations in Einstein-Inspired Notation for Python.

A reactive notebook for Python — run reproducible experiments, execute as a script, deploy as an app, and version with git.

A dead simple Python package for creating custom JAX pytree objects

A functional training loops library for JAX

A massively parallel, high-level programming language

🔮 ChatGPT Desktop Application (Mac, Windows and Linux)

📘 dict subclass with keylist/keypath support, built-in I/O operations (base64, csv, html, ini, json, pickle, plist, query-string, toml, xls, xml, yaml), s3 support and many utilities.

JAX implementation of RL algorithms and vectorized environments

Lenia - Mathematical Life Forms