Algomancer / Bayesian-Flow-Networks

Related projects: ⓘ

• nnaisense / bayesian-flow-networks
  This is the official code release for Bayesian Flow Networks.
  ☆234Updated 2 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
• kuleshov-group / mdlm
  Simplified Masked Diffusion Language Model
  ☆160Updated 2 weeks ago
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆205Updated 8 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
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆154Updated 11 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆118Updated 9 months ago
• lucidrains / rectified-flow-pytorch
  Implementation of rectified flow and some of its followup research / improvements in Pytorch
  ☆141Updated 3 weeks ago
• gle-bellier / flow-matching
  Annotated Flow Matching paper
  ☆91Updated this week
• NVlabs / edm2
  Analyzing and Improving the Training Dynamics of Diffusion Models (EDM2)
  ☆480Updated 3 months ago
• lucidrains / recurrent-interface-network-pytorch
  Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…
  ☆193Updated 7 months ago
• google-research / vdm
  ☆290Updated 2 years ago
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆274Updated last month
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆639Updated 4 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆62Updated last month
• cloneofsimo / scaling-guide
  WIP
  ☆76Updated last month
• wpeebles / G.pt
  Official PyTorch Implementation of "Learning to Learn with Generative Models of Neural Network Checkpoints"
  ☆335Updated last year
• lucidrains / attention
  ☆192Updated this week
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆445Updated 6 months ago
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
  ☆126Updated this week
• soniajoseph / ViT-Prisma
  ViT Prisma is a mechanistic interpretability library for Vision Transformers (ViTs).
  ☆161Updated this week
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆84Updated 4 months ago
• apple / ml-sigma-reparam
  ☆288Updated 2 months ago
• BlackHC / neural_net_checklist
  ☆137Updated last month
• Newbeeer / pfgmpp
  Code for ICML 2023 paper, "PFGM++: Unlocking the Potential of Physics-Inspired Generative Models"
  ☆354Updated last year
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆140Updated last year
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. More efficient than using for loops, but probably less efficient tha…
  ☆89Updated 5 months ago
• ehoogeboom / multinomial_diffusion
  ☆199Updated last year
• google-deepmind / nanodo
  ☆180Updated 2 months ago