A fork of OpenAI Baselines, implementations of reinforcement learning algorithms

Ongoing research training transformer language models at scale, including: BERT & GPT-2

This is an official implementation for "Video Swin Transformers".

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation.

Quantized inference code for LLaMA models

Neural MMO - A Massively Multiagent Environment for Artificial Intelligence Research

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

ONNX-compatible LightGlue: Local Feature Matching at Light Speed. Supports TensorRT, OpenVINO

Implementation of reinforcement learning approach to make a car learn to drive smoothly in minutes

A Large Quadcopter Swarm

WebUI extension for ControlNet, supports LoRA version of ControlNet

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch - fork with video pseudo3d

Implementation of Stochastic Beam Search using Fairseq

Code for "CosyPose: Consistent multi-view multi-object 6D pose estimation", ECCV 2020.

This repo contains the code and configuration files for reproducing object detection results of FocalNets with DINO

Driving in CARLA using model-free deep reinforcement learning

ROS packages for Turtlebot3

JAX implementation ViT-VQGAN

Mask3D predicts accurate 3D semantic instances achieving state-of-the-art on ScanNet, ScanNet200, S3DIS and STPLS3D.

An OpenAI Gym environment for the Flappy Bird game

An object-based toolbox for robot dynamic simulation, analysis, control and planning.

MATE: the Multi-Agent Tracking Environment.

The implementation for DropMessage.

Crosswalks and traffic lights detection for people with visual impairment and blindness using deep learning.

PyTorch implementation of "Towards k-means-friendly spaces: Simultaneous deep learning and clustering," Bo Yang et al., 2017.

2D car simulator for competition.