[ICLR 2025] Robust Gymnasium: A Unified Modular Benchmark for Robust Reinforcement Learning.

Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" (ICML 2024).

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

Implementation of Dreamer v3 in pytorch.

Machine Learning library for the emerging Mojo/Python ecosystem

Statistical Rethinking with PyTorch and Pyro

NeurIPS 2023: Safety-Gymnasium: A Unified Safe Reinforcement Learning Benchmark

[ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and use planning (Dyna-MPC) during fine-tuning.

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

The Controllable Agent project trains RL Agents able to optimize any reward function specified in real time, without any further learning or fine-tuning. Training is reward-free and based on the Fo…

DL4TO is a Python library for 3D topology optimization that is based on PyTorch and allows easy integration with neural networks.

DMControl Generalization Benchmark

Learnable latent embeddings for joint behavioral and neural analysis - Official implementation of CEBRA

Contrib package for Stable-Baselines3 - Experimental reinforcement learning (RL) code

Lightweight multi-agent gridworld Gym environment

[NeurIPS 2019] Code for the paper "Learning to Control Self-Assembling Morphologies: A Study of Generalization via Modularity"

Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.

Deep active inference agents using Monte-Carlo methods

Disentanglement library for PyTorch

Implementation of 'RL^2: Fast Reinforcement Learning via Slow Reinforcement Learning'

Animal-AI 3

Unofficial pytorch implementation of 'Image Inpainting for Irregular Holes Using Partial Convolutions' [Liu+, ECCV2018]

Probabilistic Machine Learning: Advanced Topics

This repository contains code for applying Riemannian geometry in machine learning.

A curated collection of resources and research related to the geometry of representations in the brain, deep networks, and beyond

Flexible modulation of sequence generation in the entorhinal-hippocampal system.

TikZ library for drawing Bayesian networks, graphical models and (directed) factor graphs in LaTeX.

Accepted in MIDL 2020

CPAB Transformations: finite-dimensional spaces of simple, fast, and highly-expressive diffeomorphisms derived from parametric, continuously-defined, velocity fields in Numpy, Tensorflow and Pytorch