Alternatives and similar repositories for micro_diffusion

Users that are interested in micro_diffusion are comparing it to the libraries listed below

• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆651Updated last year
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,228Updated 5 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆613Updated last year
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆643Updated 4 months ago
• policy-gradient / GRPO-Zero
  Implementing DeepSeek R1's GRPO algorithm from scratch
  ☆1,638Updated 6 months ago
• NVIDIA / Cosmos-Tokenizer
  A suite of image and video neural tokenizers
  ☆1,675Updated 8 months ago
• apple / ml-4m
  4M: Massively Multimodal Masked Modeling
  ☆1,767Updated 5 months ago
• helblazer811 / Diffusion-Explorer
  Interactive visualizations of the geometric intuition behind diffusion models.
  ☆837Updated 4 months ago
• huggingface / picotron
  Minimalistic 4D-parallelism distributed training framework for education purpose
  ☆1,873Updated 2 months ago
• KellerJordan / modded-nanogpt
  NanoGPT (124M) in 3 minutes
  ☆3,755Updated this week
• KellerJordan / Muon
  Muon is an optimizer for hidden layers in neural networks
  ☆1,960Updated 3 months ago
• facebookresearch / blt
  Code for BLT research paper
  ☆2,004Updated 5 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆948Updated last year
• pytorch / ao
  PyTorch native quantization and sparsity for training and inference
  ☆2,464Updated this week
• apple / ml-mdm
  Train high-quality text-to-image diffusion models in a data & compute efficient manner
  ☆509Updated 7 months ago
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆561Updated 3 months ago
• microsoft / Samba
  [ICLR 2025] Samba: Simple Hybrid State Space Models for Efficient Unlimited Context Language Modeling
  ☆917Updated 6 months ago
• EurekaLabsAI / micrograd
  The Autograd Engine
  ☆662Updated last year
• pytorch / torchtitan
  A PyTorch native platform for training generative AI models
  ☆4,604Updated last week
• EurekaLabsAI / tensor
  The Tensor (or Array)
  ☆452Updated last year
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆526Updated last week
• NousResearch / DisTrO
  Distributed Training Over-The-Internet
  ☆963Updated 2 weeks ago
• apple / ml-aim
  This repository provides the code and model checkpoints for AIMv1 and AIMv2 research projects.
  ☆1,380Updated 2 months ago
• facebookresearch / perception_models
  State-of-the-art Image & Video CLIP, Multimodal Large Language Models, and More!
  ☆1,718Updated last month
• FoundationVision / LlamaGen
  Autoregressive Model Beats Diffusion: 🦙 Llama for Scalable Image Generation
  ☆1,882Updated last year
• srush / Triton-Puzzles
  Puzzles for learning Triton
  ☆2,090Updated 11 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆516Updated 2 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆294Updated last year
• meta-pytorch / torchcodec
  PyTorch media decoding and encoding
  ☆782Updated this week
• huggingface / nanoVLM
  The simplest, fastest repository for training/finetuning small-sized VLMs.
  ☆4,205Updated last week