Medusa: Simple Framework for Accelerating LLM Generation with Multiple Decoding Heads

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

Fully open reproduction of DeepSeek-R1

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Official repo for paper "Structured 3D Latents for Scalable and Versatile 3D Generation".

Cosmos is a world model development platform that consists of world foundation models, tokenizers and video processing pipeline to accelerate the development of Physical AI at Robotics & AV labs. C…

HunyuanVideo: A Systematic Framework For Large Video Generation Model

[ICLR'25] Official Implementation for Consistent Flow Distillation for Text-to-3D Generation

A library for differentiable robotics.

Machine Learning Engineering Open Book

[ECCV2024 - Oral, Best Paper Award Candidate] SEA-RAFT: Simple, Efficient, Accurate RAFT for Optical Flow

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

You See it, You Got it: Learning 3D Creation on Pose-Free Videos at Scale

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.

Code for the project "MegaSaM: Accurate, Fast and Robust Structure and Motion from Casual Dynamic Videos"

Easily run PyTorch on multiple GPUs & machines

Code for the manim-generated scenes used in 3blue1brown videos

MoGe: Unlocking Accurate Monocular Geometry Estimation for Open-Domain Images with Optimal Training Supervision

Estimating Body and Hand Motion in an Ego-sensed World

Official Implementation of paper "MonST3R: A Simple Approach for Estimating Geometry in the Presence of Motion"

The best OSS video generation models

For an education purpose, from-scratch, single-file, python-only pose-graph optimization implementation

Code of Pyramidal Flow Matching for Efficient Video Generative Modeling

Implement a ChatGPT-like LLM in PyTorch from scratch, step by step

Bring portraits to life!

💤 A modern plugin manager for Neovim

A plugin to visualise and resolve merge conflicts in neovim

Use your Neovim like using Cursor AI IDE!