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
- Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.☆13Jun 24, 2023Updated 2 years ago
- A collection of lectures and tutorials☆24Jun 14, 2024Updated last year
- JAX-based linear Einstein-Boltzmann solver for cosmology☆24Nov 13, 2025Updated 3 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
- ☆22Aug 11, 2023Updated 2 years ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆97Feb 12, 2024Updated 2 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
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 3 years ago
- Official code repository of the paper Learning Associative Inference Using Fast Weight Memory by Schlag et al.☆29Feb 25, 2021Updated 5 years ago
- 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
- Differentiable cosmological emulators: the JAX version of CosmoPower☆44Nov 27, 2025Updated 3 months ago
- PyTorch-based radio-interferometric imaging reconstruction package with scalable Bayesian uncertainty quantification relying on data-driv…☆12Feb 17, 2025Updated last year
- PyTorch Implementation of V-objective Diffusion Probabilistic Models with Classifier-free Guidance☆35Mar 18, 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
- ☆43May 3, 2024Updated last year
- 3rd placed submission to the NeurIPS MineRL competition 2019☆10Mar 24, 2023Updated 2 years ago
- Implementation of "Face detection in untrained deep neural networks" (Baek et al., Nature Communications, 2021)☆10Nov 2, 2021Updated 4 years ago
- Testing Difference Target Propagation (DTP) on MNIST.☆12Oct 12, 2020Updated 5 years ago
- Automatize local data analysis with team of tool-using GPT agents☆15Apr 1, 2024Updated last year
- ☆19Nov 25, 2024Updated last year
- PyTorch reimplementation of Noise2Same with enhancements☆11Nov 9, 2024Updated last year
- Matlab functions to read and write SVG (Scalable Vector Graphic) files to and from Matlab structures☆11Jan 29, 2023Updated 3 years ago
- Analytical methods for efficient inference of integrate-and-fire circuit models from single-trial spike trains☆10Oct 30, 2019Updated 6 years ago
- Tensorflow implementation of "MC2-Net: Motion Correction Network for Multi-Contrast Brain MRI"☆12Dec 8, 2022Updated 3 years ago
- An open source reinforcement learning codebase with a variety of intrinsic exploration methods implemented in PyTorch.☆11Feb 6, 2023Updated 3 years ago
- a differentiable lensing simulator☆10Jan 18, 2022Updated 4 years ago
- tutorials on shape measurement☆10Jan 5, 2022Updated 4 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated 5 months ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- Facial Recognition Software for Macaque Monkeys☆13Jul 18, 2017Updated 8 years ago
- [ICML 2024]: Official implementation for the paper: "Consistent Diffusion Meets Tweedie"☆53Apr 26, 2024Updated last year
- NumPy-style histograms in PyTorch☆54Sep 17, 2025Updated 5 months ago