bhyang

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

dspy

DSPy: The framework for programming—not prompting—foundation models

pix2pix

Image-to-image translation with conditional adversarial nets

python-machine-learning-book-2nd-edition

The "Python Machine Learning (2nd edition)" book code repository and info resource

torch-rnn

Efficient, reusable RNNs and LSTMs for torch

muzero-general

MuZero

d3rlpy

An offline deep reinforcement learning library

EfficientZero

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

argoverse-api

Official GitHub repository for Argoverse dataset

purejaxrl

Really Fast End-to-End Jax RL Implementations

Solid

🎯 A comprehensive gradient-free optimization framework written in Python

NPEET

Non-parametric Entropy Estimation Toolbox

nocturne

A data-driven, fast driving simulator for multi-agent coordination under partial observability.

dreamerv2

Pytorch implementation of Dreamer-v2: Visual Model Based RL Algorithm.

Diffusion-Policies-for-Offline-RL

CX_DB8

a contextual, biasable, word-or-sentence-or-paragraph extractive summarizer powered by the latest in text embeddings (Bert, Universal Sentence Encoder, Flair)

plant

[CoRL'22] PlanT: Explainable Planning Transformers via Object-Level Representations

WorldOnRails

(ICCV 2021, Oral) RL and distillation in CARLA using a factorized world model

deep_bisim4control

Learning Invariant Representations for Reinforcement Learning without Reconstruction

WIMP

[arXiv] What-If Motion Prediction for Autonomous Driving ❓🚗💨

dreamer-torch

Pytorch version of Dreamer, which follows the original TF v2 codes.

AutoBots

mpo

PyTorch Implementation of the Maximum a Posteriori Policy Optimisation

LearningByCheating

Driving in CARLA using model-free deep reinforcement learning

V-MPO_Lunarlander

Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238

bayesian_optimization

Bayesian optimization

LOOP

Learning Off-Policy with Online Planning [CoRL 2021 Best Paper Finalist]

arxiv-filter

reminiscent_tracker

flocking-swarmAI