Lists (2)
Stars
Fast and memory-efficient exact attention
Functional local implementations of main model parallelism approaches
PyTorch extensions for high performance and large scale training.
The simplest, fastest repository for training/finetuning medium-sized GPTs.
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training
GFlowNet library specialized for graph & molecular data
C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.
Train transformer language models with reinforcement learning.
Code and links for over 25,000 trained Atari agents
A personal project where I publish my research paper notes on a weekly basis.
Editorial venue for ML Reproducibility Challenge Accepted papers
A library to inspect and extract intermediate layers of PyTorch models.
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Determined is an open-source machine learning platform that simplifies distributed training, hyperparameter tuning, experiment tracking, and resource management. Works with PyTorch and TensorFlow.
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
A collection of 100+ pre-trained RL agents using Stable Baselines, training and hyperparameter optimization included.
A programming language to skip the things you have already computed
IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation
Options of Interest: Temporal Abstraction with Interest Functions AAAI 2020
Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.
Official code for ICLR 2020 paper "A Neural Dirichlet Process Mixture Model for Task-Free Continual Learning."