smsharma / minified-generative-models
Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc
☆128Updated last year
Alternatives and similar repositories for minified-generative-models:
Users that are interested in minified-generative-models are comparing it to the libraries listed below
- Collecting research materials on neural samplers with diffusion/flow models☆46Updated last month
- Official Jax Implementation of MD4 Masked Diffusion Models☆77Updated last month
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- Flow-matching algorithms in JAX☆89Updated 8 months ago
- Code for the paper https://arxiv.org/abs/2402.04997☆77Updated last year
- code for "Riemannian Flow Matching on General Geometries".☆219Updated last year
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆51Updated 2 weeks ago
- ☆106Updated 11 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 8 months ago
- Diffusion models in PyTorch☆101Updated last month
- Simple Guidance Mechanisms for Discrete Diffusion Models☆31Updated 4 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems☆45Updated last week
- Educational implementation of the Discrete Flow Matching paper☆85Updated 7 months ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated last year
- ☆141Updated last year
- Graph neural networks in JAX.☆67Updated 10 months ago
- Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]☆56Updated last month
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last month
- Run PyTorch in JAX. 🤝☆236Updated 2 months ago
- Implementation of Action Matching☆41Updated last year
- A short and easy PyTorch implementation of E(n) Equivariant Graph Neural Networks☆131Updated 3 years ago
- Visualizations of the theory behind diffusion models.☆187Updated last year
- Unofficial implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch☆59Updated 2 weeks ago
- Pytorch-like dataloaders for JAX.☆80Updated this week
- Discovering Interpretable Features in Protein Language Models via Sparse Autoencoders☆168Updated 2 months ago
- Official implementation of the NeurIPS 24 paper of statistical flow matching (SFM) for discrete generation.☆24Updated 5 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆52Updated last month
- Code repository for Trajectory Flow Matching☆62Updated 5 months ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆49Updated last month
- This is the official code release for Bayesian Flow Networks.☆272Updated 9 months ago