Ring attention implementation with flash attention

Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.

Databricks’ Dolly, a large language model trained on the Databricks Machine Learning Platform

Drive a browser with Cohere

This is a ZSH plugin that enables you to use OpenAI's Codex AI in the command line.

Search the most relevant emojis given a natural language query

CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery

PyTorch Code for "Generalization in Dexterous Manipulation via Geometry-Aware Multi-Task Learning"

Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL

A PyTorch implementation of Implicit Behavioral Cloning

Code for the paper "Batch size invariance for policy optimization"

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.

Development repository for the Triton language and compiler

DrQ-v2: Improved Data-Augmented Reinforcement Learning

Open source code for AlphaFold 2.

RE3: State Entropy Maximization with Random Encoders for Efficient Exploration

Render OpenAI Gym environments in Google Colaboratory

Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.

RL research on Android devices.

An enterprise-class UI design language and React UI library

The offical notes of Andrew Ng Machine Learning in Stanford University

Library for Model Based RL

A minimal implementation of Go-Explore without domain knowledge

FAIR Chemistry's library of machine learning methods for chemistry

COVID deterioration prediction based on chest X-ray radiographs via MoCo-trained image representations

Reinforcement Learning with Latent Flow