Alternatives and similar repositories for DiffuLab

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

• evanatyourservice / kron_torch
  An implementation of PSGD Kron second-order optimizer for PyTorch
  ☆98Updated 6 months ago
• facebookresearch / capi
  Code and weights for the paper "Cluster and Predict Latents Patches for Improved Masked Image Modeling"
  ☆129Updated 2 weeks ago
• lucidrains / evolutionary-policy-optimization
  Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University
  ☆103Updated 4 months ago
• galilai-group / lejepa
  ☆851Updated this week
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆105Updated last year
• m1balcerak / EnergyMatching
  [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"
  ☆196Updated 3 months ago
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables the optimization of neural networks with multiple losses (e.g. multi-task learning)…
  ☆296Updated this week
• thinking-machines-lab / manifolds
  Supporting code for the blog post on modular manifolds.
  ☆113Updated 4 months ago
• kvfrans / splus
  ☆123Updated 7 months ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆71Updated 2 months ago
• p-doom / jasmine
  A simple, performant and scalable JAX-based world modeling codebase.
  ☆123Updated 2 weeks ago
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆133Updated last month
• fal-ai / diffusion-speedrun
  Focused on fast experimentation and simplicity
  ☆80Updated last year
• lorenmt / clarity-template
  Clarity: A Minimalist Website Template for AI Research
  ☆177Updated last year
• google-research / kauldron
  Modular, scalable library to train ML models
  ☆203Updated this week
• lucidrains / vit-arc-slot
  Explorations into improving ViTArc with Slot Attention
  ☆43Updated last year
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆90Updated last year
• Photoroom / datago
  A natively parallel dataloader for Python, written in Rust. Serving data at GB/s speeds, while covering aspect ratio bucketing, crop and …
  ☆126Updated last week
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆29Updated last year
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆107Updated 3 months ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated last year
• Chrixtar / SRM
  [ICML 2025] Implementation of Spatial Reasoning with Denoising Models
  ☆86Updated 6 months ago
• idiap / sigma-gpt
  σ-GPT: A New Approach to Autoregressive Models
  ☆70Updated last year
• lucidrains / improving-transformers-world-model-for-rl
  Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch
  ☆151Updated 8 months ago
• BlackHC / neural_net_checklist
  ☆215Updated last year
• vmicheli / delta-iris
  Efficient World Models with Context-Aware Tokenization. ICML 2024
  ☆115Updated last year
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆38Updated last year
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆114Updated last year
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆42Updated 8 months ago
• perceptron-ai-inc / perceptron
  The official Python SDK for the Perceptron API
  ☆58Updated last week