Implementation of Dreamer v3 in pytorch.

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax

Tensor library for machine learning

Scikit-learn implemented with PyTorch

A Gym Interface for Multi-Task Reinforcement Learning.

Alternative Twitter front-end

Read Google Cloud Storage, Azure Blobs, and local paths with the same interface

Holistic Evaluation of Language Models (HELM), a framework to increase the transparency of language models (https://arxiv.org/abs/2211.09110). This framework is also used to evaluate text-to-image …

Mastering Diverse Domains through World Models

Port of OpenAI's Whisper model in C/C++

Lightning fast C++/CUDA neural network framework

State Space Models library in JAX

LaTeX template in accordance with the University of Southern California Theses and Dissertations Formatting Guidelines.

A little Python script to collect LaTeX sources for upload to the arXiv.

A modular RL library to fine-tune language models to human preferences

This project aims to create a crystal clear tutorial on a cryptic looking topic.

High-quality single-file implementations of SOTA Offline and Offline-to-Online RL algorithms: AWAC, BC, CQL, DT, EDAC, IQL, SAC-N, TD3+BC, LB-SAC, SPOT, Cal-QL, ReBRAC

Tax filing web application

A Smart Ethernet Switch for Earth

Sync changes in Paperpile to a Notion database.

ttf-opensans is a Python package vending the Open Sans font by Steve Matteson.

An open access book on scientific visualization using python and matplotlib

Extending JAX with custom C++ and CUDA code

A declarative drawing API in Python

A reinforcement learning package for Julia

Best Practices on Recommendation Systems

Code and links for over 25,000 trained Atari agents

A C++ pytorch implementation of MuZero