yuanchenyang / smalldiffusion

Related projects: ⓘ

• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆246Updated 3 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆236Updated 8 months ago
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆69Updated 7 months ago
• google-deepmind / nanodo
  ☆180Updated 2 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆316Updated last month
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆639Updated 4 months ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆63Updated 5 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆187Updated 10 months ago
• louaaron / Score-Entropy-Discrete-Diffusion
  [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)
  ☆352Updated 6 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆168Updated 4 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆503Updated this week
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆411Updated 2 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆445Updated 6 months ago
• cabralpinto / modular-diffusion
  Python library for designing and training your own Diffusion Models with PyTorch.
  ☆266Updated last month
• KellerJordan / modded-nanogpt
  GPT-2 (124M) quality in 5B tokens
  ☆227Updated last week
• FlorianDietz / comgra
  A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…
  ☆228Updated this week
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆240Updated last week
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆205Updated 8 months ago
• eloialonso / diamond
  DIAMOND (DIffusion As a Model Of eNvironment Dreams) is a reinforcement learning agent trained in a diffusion world model.
  ☆206Updated last week
• apple / ml-sigma-reparam
  ☆288Updated 2 months ago
• VSehwag / minimal-diffusion
  A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)
  ☆244Updated 2 weeks ago
• BlackHC / neural_net_checklist
  ☆137Updated last month
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆124Updated 3 weeks ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆224Updated last year
• kuleshov-group / mdlm
  Simplified Masked Diffusion Language Model
  ☆160Updated last week
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆155Updated 7 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆116Updated last month
• tmabraham / ddpo-pytorch
  Reproduction of DDPO paper (RLHF for diffusion)
  ☆70Updated last year
• mlcommons / algorithmic-efficiency
  MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…
  ☆321Updated 2 weeks ago
• bhosmer / mm
  ☆110Updated 2 months ago