alito

whisper

Robust Speech Recognition via Large-Scale Weak Supervision

segment-anything

The repository provides code for running inference with the SegmentAnything Model (SAM), links for downloading the trained model checkpoints, and example notebooks that show how to use the model.

taichi

Productive, portable, and performant GPU programming in Python.

audiocraft

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable music generation LM with textual and melodic conditioning.

unilm

Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities

latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

ml-engineering

Machine Learning Engineering Open Book

qlora

QLoRA: Efficient Finetuning of Quantized LLMs

FlexGen

Running large language models on a single GPU for throughput-oriented scenarios.

OpenChatKit

metaseq

Repo for external large-scale work

bitsandbytes

Accessible large language models via k-bit quantization for PyTorch.

kaitai_struct

Kaitai Struct: declarative language to generate binary data parsers in C++ / C# / Go / Java / JavaScript / Lua / Nim / Perl / PHP / Python / Ruby

timesfm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.

textual_inversion

Eureka

Official Repository for "Eureka: Human-Level Reward Design via Coding Large Language Models" (ICLR 2024)

LongLoRA

Code and documents of LongLoRA and LongAlpaca (ICLR 2024 Oral)

diplomacy_cicero

Code for Cicero, an AI agent that plays the game of Diplomacy with open-domain natural language negotiation.

iTransformer

Official implementation for "iTransformer: Inverted Transformers Are Effective for Time Series Forecasting" (ICLR 2024 Spotlight), https://openreview.net/forum?id=JePfAI8fah

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control

Text2LIVE

Official Pytorch Implementation for "Text2LIVE: Text-Driven Layered Image and Video Editing" (ECCV 2022 Oral)

iris

Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.

DeepDPM

"DeepDPM: Deep Clustering With An Unknown Number of Clusters" [Ronen, Finder, and Freifeld, CVPR 2022]

rl_with_resets

JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"

director

Deep Hierarchical Planning from Pixels

METRA

METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)

omni

OMNI: Open-endedness via Models of human Notions of Interestingness

lattice

[NeurIPS 2023] Latent Exploration for Reinforcement Learning

AtMan

PPO

Example implemention of the Proximal Policy Optimization algorithm