Stars
OpenMMLab Detection Toolbox and Benchmark
[NeurIPS'23 Oral] Visual Instruction Tuning (LLaVA) built towards GPT-4V level capabilities and beyond.
OpenAI Baselines: high-quality implementations of reinforcement learning algorithms
An open source implementation of CLIP.
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
A paper list of object detection using deep learning.
High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)
Official repository of "SAMURAI: Adapting Segment Anything Model for Zero-Shot Visual Tracking with Motion-Aware Memory"
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
Model interpretability and understanding for PyTorch
PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKT…
An implementation of the AlphaZero algorithm for Gomoku (also called Gobang or Five in a Row)
Modularized Implementation of Deep RL Algorithms in PyTorch
Implementations of basic RL algorithms with minimal lines of codes! (pytorch based)
Official codebase for Decision Transformer: Reinforcement Learning via Sequence Modeling.
Isaac Gym Reinforcement Learning Environments
Simple and easily configurable grid world environments for reinforcement learning
Python Multi-Agent Reinforcement Learning framework
DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model. NeurIPS 2024 Spotlight.
Rainbow: Combining Improvements in Deep Reinforcement Learning
Code for the paper "Neural MMO: A Massively Multiagent Game Environment for Training and Evaluating Intelligent Agents"
bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent
Minimalistic 4D-parallelism distributed training framework for education purpose
Clean PyTorch implementations of imitation and reward learning algorithms
Video wallpaper for Linux. Written in Python. 🐍
[ICML 2017] TensorFlow code for Curiosity-driven Exploration for Deep Reinforcement Learning
[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCTS)
Theano-based implementation of Deep Q-learning
Code for the paper "Exploration by Random Network Distillation"
A PyTorch Platform for Distributed RL