Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.
☆19Jun 11, 2023Updated 2 years ago
Alternatives and similar repositories for consistency-models
Users that are interested in consistency-models are comparing it to the libraries listed below
Sorting:
- Unofficial Implementation of Consistency Models in pytorch☆260Mar 7, 2023Updated 2 years ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 9 months ago
- A mini-library for training consistency models.☆251Dec 26, 2023Updated 2 years ago
- MLX implementation of GCN, with benchmark on MPS, CUDA and CPU (M1 Pro, M2 Ultra, M3 Max).☆25Dec 16, 2023Updated 2 years ago
- Self-Similarity Priors: Neural Collages as Differentiable Fractal Representations☆29Nov 26, 2022Updated 3 years ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆97Feb 12, 2024Updated 2 years ago
- Graph neural networks tutorial in pytorch (GCN, GAT, Node2vec, GraphSAge, ClusterGCN, ...)☆26Jan 20, 2022Updated 4 years ago
- Spiking neural networks in JAX with equinox.☆29Aug 29, 2024Updated last year
- Simulation-based (likelihood-free) inference customized for astronomical applications☆29Nov 4, 2024Updated last year
- Unofficial implementation of EDM2: Analyzing and Improving the Training Dynamics of Diffusion Models☆12Mar 17, 2024Updated last year
- T2N is an extension of the TREES toolbox providing an interface between Matlab and the compartmental modeling environment NEURON. It w…☆10Sep 3, 2019Updated 6 years ago
- A JAX implementation of the continuous time formulation of Consistency Models☆85Apr 7, 2023Updated 2 years ago
- Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization☆34Oct 31, 2025Updated 4 months ago
- 1-D Toy Diffusion model written in pure Pytorch☆29Sep 12, 2022Updated 3 years ago
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- A collection of graph neural networks implementations in JAX☆35Nov 28, 2023Updated 2 years ago
- A Toolkit for OpenAI's Consistency Models.☆207Apr 14, 2023Updated 2 years ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- Implementation of Mejias et al. 2016: Feedforward and feedback frequency-dependent interactions in a large-scale laminar network of the p…☆10Nov 25, 2025Updated 3 months ago
- PyTorch Implementation of V-objective Diffusion Probabilistic Models with Classifier-free Guidance☆35Mar 18, 2024Updated last year
- Find strongest response of convolutional layers on an image dataset. Automatically compute receptive field for any CNN layer.☆14Feb 19, 2021Updated 5 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆55Jul 15, 2025Updated 7 months ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- Runway Port of BigBiGAN from the paper "Large Scale Adversarial Representation Learning"☆10Dec 10, 2024Updated last year
- Mymou: A standalone touchscreen in-cage testing device☆17Nov 25, 2024Updated last year
- Implicit Distributional Actor Critic☆11Dec 8, 2021Updated 4 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 5 months ago
- [NeurIPS 2024 Spotlight] code for "Diffusion Model with Cross Attention as an Inductive Bias for Disentanglement"☆18Jan 26, 2025Updated last year
- ☆10Oct 27, 2023Updated 2 years ago
- Implementation of the ICLR 2022 paper "Phase Collapse in Neural Networks."☆10Mar 21, 2022Updated 3 years ago
- Tensorflow implementation of "MC2-Net: Motion Correction Network for Multi-Contrast Brain MRI"☆12Dec 8, 2022Updated 3 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- Automatize local data analysis with team of tool-using GPT agents☆15Apr 1, 2024Updated last year
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- Python implementation of Predictive Coding Networks and Predictive Coding Graphs, using PyTorch.☆17Jul 22, 2024Updated last year
- Facial Recognition Software for Macaque Monkeys☆13Jul 18, 2017Updated 8 years ago
- Testing Difference Target Propagation (DTP) on MNIST.☆12Oct 12, 2020Updated 5 years ago
- Learning to draw samples: with application to amortized maximum likelihood estimator for generative adversarial learning☆10Dec 28, 2021Updated 4 years ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year