3D-DLP: Self-supervised 3D Object-centric Scene Representation Learning (ICML 2026)

[ICLR 2026 Oral] Latent Particle World Models official repository

Official PyTorch implementation of "Hierarchical Entity-centric Reinforcement Learning with Factored Subgoal Diffusion", Haramati et al., ICLR 2026

SR-DiT Speedrunning ImageNet Diffusion

A minimal implementation of DeepMind's Genie world model

Official implementation of EC-Diffuser: Multi-Object Manipulation via Entity-Centric Behavior Generation

Official PyTorch implementation of "Entity-Centric Reinforcement Learning for Object Manipulation from Pixels", Haramati et al., ICLR 2024

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

This Python library makes it easy to display images and videos in a notebook.

A community-maintained Python framework for creating mathematical animations.

A small, interpretable codebase containing the re-implementation of a few "deep" NLP models in PyTorch. Colab notebooks to run with GPUs. Models: word2vec, CNNs, transformer, gpt.

Official code for the paper "Semantically Driven Sentence Fusion: Modeling and Evaluation".

The goal of this library is to generate more helpful exception messages for matrix algebra expressions for numpy, pytorch, jax, tensorflow, keras, fastai.

Mastering Blackjack - Combining Supervised Learning with Deep Reinforcement Learning

Python3 ROS Interface with the Franka Emika Panda robot arm. Includes support for OpenAI Gym based interfaces.

[CVPR2020] Adversarial Latent Autoencoders

S-RL Toolbox: Reinforcement Learning (RL) and State Representation Learning (SRL) for Robotics

A PyTorch re-implementation of Neural Radiance Fields

Simple and easily configurable grid world environments for reinforcement learning

Information Distillation Generative Adversrial Network in PyTorch

Object representation learning

A pytorch implementation of Paper "IntroVAE: Introspective Variational Autoencoders for Photographic Image Synthesis"

PyTorch Agent Net: reinforcement learning toolkit for pytorch

Sequence to sequence learning using TensorFlow.