junmokane

torchgfn

GFlowNet library

diffusion-finetuning

rlbase_stable

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

gfn-lm-tuning

sysbinder

Official Code for Neural Systematic Binder

parti-pytorch

Implementation of Parti, Google's pure attention-based text-to-image neural network, in Pytorch

GenSim

GenSim: Generating Robotic Simulation Tasks via Large Language Models

robocasa

RoboCasa: Large-Scale Simulation of Everyday Tasks for Generalist Robots

mini_edm

Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist

denoising-diffusion-pytorch

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

latent-slot-diffusion

Official Release of NeurIPS 2023 Spotlight paper "Object-Centric Slot Diffusion"

latent-diffusion

High-Resolution Image Synthesis with Latent Diffusion Models

mamba.py

A simple and efficient Mamba implementation in pure PyTorch and MLX.

zeta

Build high-performance AI models with modular building blocks

VisionMamba

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

unleashing-transformers

Code for the ECCV 2022 paper "Unleashing Transformers"

MaskGIT-pytorch

Pytorch implementation of MaskGIT: Masked Generative Image Transformer (https://arxiv.org/pdf/2202.04200.pdf)

dreamerv3-torch

Implementation of Dreamer v3 in pytorch.

dreamerv3

Mastering Diverse Domains through World Models

disentangling-vae

Experiments for understanding disentanglement in VAE latent representations

enhancing-transformers

An unofficial implementation of both ViT-VQGAN and RQ-VAE in Pytorch

pytorch-a2c-ppo-acktr-gail

PyTorch implementation of Advantage Actor Critic (A2C), Proximal Policy Optimization (PPO), Scalable trust-region method for deep reinforcement learning using Kronecker-factored approximation (ACKTR) and Generative Adversarial Imitation Learning (GAIL).

steve

Official code for Slot-Transformer for Videos (STEVE)

rl-paper-review

road-map & paper review for Reinforcement Learning

Model_Generalization_Code_supplement

Code for "The Benefits of Model-Based Generalization in Reinforcement Learning"

OCRL

Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to RL tasks.

thortils

Utility functions when working with Ai2-THOR. Try to do one thing once.

procthor

🏘️ Scaling Embodied AI by Procedurally Generating Interactive 3D Houses

Object2HabitatMap

Awesome habitat top down map work 🤩