Neural Turing Machines (NTM) - PyTorch Implementation

Ultralytics YOLO11 🚀

SBX: Stable Baselines Jax (SB3 + Jax)

Implementation of the POIS algorithm

Official implementation for the paper "Offline Meta RL - Identifiability Challenges and Effective Data Collection Strategies", NeurIPS 2021.

Massively parallel rigidbody physics simulation on accelerator hardware.

An API standard for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

Aruco Marker Calibration and Pose Estimation using OpenCV in Python

Acceptance rates for the major AI conferences

Code for Stabilizing Off-Policy RL via Bootstrapping Error Reduction

Code for conservative Q-learning

The AI developer platform. Use Weights & Biases to train and fine-tune models, and manage models from experimentation to production.

Count the MACs / FLOPs of your PyTorch model.

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

An offline deep reinforcement learning library

An elegant PyTorch offline reinforcement learning library for researchers.

Official code for "RAMBO: Robust Adversarial Model-Based Offline RL", NeurIPS 2022

The unofficial LaTeX Beamer template for presentation slides at Singapore Management University (SMU).

A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities

Streamlining reinforcement learning with RLOps. State-of-the-art RL algorithms and tools.

Softlearning is a reinforcement learning framework for training maximum entropy policies in continuous domains. Includes the official implementation of the Soft Actor-Critic algorithm.

Code for the paper Fine-Tuning Language Models from Human Preferences

ChatRWKV is like ChatGPT but powered by RWKV (100% RNN) language model, and open source.

Platform to experiment with the AI Software Engineer. Terminal based. NOTE: Very different from https://gptengineer.app

A RWKV management and startup tool, full automation, only 8MB. And provides an interface compatible with the OpenAI API. RWKV is a large language model that is fully open source and available for c…

RWKV is an RNN with transformer-level LLM performance. It can be directly trained like a GPT (parallelizable). So it's combining the best of RNN and transformer - great performance, fast inference,…

A toolkit for developing and comparing reinforcement learning algorithms.

Inpaint anything using Segment Anything and inpainting models.

[NeurIPS 22] [AAAI 24] Recurrent Transformer-based long-context architecture.

Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch