PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

Code release for ConvNeXt model

Official PyTorch Implementation of "SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers"

[ICLR'25 Oral] Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

Tracking Any Point (TAP)

Fully open reproduction of DeepSeek-R1

A generative world for general-purpose robotics & embodied AI learning.

A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes practical examples for both text and image modalities.

An independent extension based on IsaacLab. It provides support for Robot Manipulation tasks (Robot Arm and Dextrous Hand).

A Versatile Teleoperation framework for Robotic Manipulation using Meta Quest3

A Python framework for high performance GPU simulation and graphics

RDT-1B: a Diffusion Foundation Model for Bimanual Manipulation

Official implementation for VIOLA

🚀🚀 「大模型」2小时完全从0训练26M的小参数GPT！🌏 Train a 26M-parameter GPT from scratch in just 2h!

Mamba SSM architecture

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RN…

Approaching (Almost) Any Machine Learning Problem

Learning-based locomotion control from OpenRobotLab, including Hybrid Internal Model & H-Infinity Locomotion Control

unitree Go2 robot learns locomotion with N-P3O algorithm and HIM alike policy trained by isaacgym

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)

CUDA Accelerated Robot Library

PyTorch code and models for the DINOv2 self-supervised learning method.

Conditional diffusion model to generate MNIST. Minimal script. Based on 'Classifier-Free Diffusion Guidance'.

Karras et al. (2022) diffusion models for PyTorch

Hackable and optimized Transformers building blocks, supporting a composable construction.

Implementation of Denoising Diffusion Probabilistic Model in Pytorch