Dreamfields

github.comsite

Official implementation of the ICLR 2022 paper Progressive Distillation for Fast Sampling of Diffusion Models.

from

JAX

added

2026-10-10

likes

0

similar

1. ▲
   flaxdiff github.com

   tool category #jax link ↗

   Framework and Library for building and training Diffusion models in multi-node multi-device distributed settings (TPUs)

2. ▲
   D3PM github.com

   site category #jax link ↗

   Official implementation of Structured Denoising Diffusion Models in Discrete State-Spaces.

3. ▲
   Light Field Neural Rendering github.com

   site category #jax link ↗

   Official implementation of Light Field Neural Rendering.

4. ▲
   DeepSeek-R1-Flax-1.5B-Distill github.com

   tool category #jax link ↗

   Flax implementation of DeepSeek-R1 1.5B distilled reasoning LLM.

   
5. ▲
   DiffeRT github.com

   tool category #jax link ↗

   Differentiable Ray Tracing toolbox for Radio Propagation powered by the JAX ecosystem.

6. ▲
   JaxDF github.com

   tool category #jax link ↗

   Framework for differentiable simulators with arbitrary discretizations.

   

JAX › Models and Projects > Flax: “Official implementation of the ICLR 2022 paper Progressive Distillation for Fast Sampling of Diffusion Models.”