consistency-models

A JAX implementation of the continuous time formulation of Consistency Models

MIT License

Stars
81
View Code on GitHub View on X
Ecosystems: Python

Issue Statistics

Past Year

All Time

Total Pull Requests
0
0
Merged Pull Requests
0
0
Total Issues
0
0
Time to Close Issues
N/A
N/A
Related Projects
k-diffusion

Karras et al. (2022) diffusion models for PyTorch

21 Jun 2022 2,277
jax-enhance

minimal library for image super-resolution implemented in jax

10 Jul 2021 10
DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

16 Dec 2022 6,050
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
mlp-gpt-jax

A GPT, made only of MLPs, in Jax

21 May 2021 55
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
mdmm-jax

Gradient-based constrained optimization for JAX

14 Sep 2022 17
pips-jax

JAX port of Persistent Independent Particles

20 Jan 2023 36
diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs....

25 Jul 2021 1,390
sympy2jax

Turn SymPy expressions into trainable JAX expressions.

23 Jun 2022 314
jax-wavelets

The 2D discrete wavelet transform for JAX

10 Feb 2023 33
v-diffusion-jax

v objective diffusion inference code for JAX.

07 Nov 2021 209
jax-comln

Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)

02 Mar 2022 27
optimistix

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.sit...

03 Oct 2023 311