piwheels

diffusionjax

diffusionjax is a simple and accessible diffusion models package in JAX

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install diffusionjax

Dependencies

• flax
• tqdm
• wandb
• matplotlib
• absl-py
• numpy
• scipy
• ml-collections

Releases

Version	Released	Bookworm Python 3.11	Trixie Python 3.13	Files
0.1.8	2025-01-19
diffusionjax-0.1.8-py3-none-any.whl (38 KB)
0.1.7	2024-01-23
diffusionjax-0.1.7-py3-none-any.whl (27 KB)
0.1.6	2024-01-19
diffusionjax-0.1.6-py3-none-any.whl (27 KB)
0.1.5	2023-10-31
diffusionjax-0.1.5-py3-none-any.whl (25 KB)
0.1.4	2023-10-26
diffusionjax-0.1.4-py3-none-any.whl (24 KB)
0.1.3	2023-10-20
diffusionjax-0.1.3-py3-none-any.whl (24 KB)
0.1.2	2023-10-10
diffusionjax-0.1.2-py3-none-any.whl (21 KB)
0.1.1	2023-09-18
diffusionjax-0.1.1-py3-none-any.whl (21 KB)
0.1.0	2023-08-22
diffusionjax-0.1.0-py3-none-any.whl (19 KB)
0.0.9	2023-05-11
diffusionjax-0.0.9-py3-none-any.whl (15 KB)
0.0.8	2023-05-10
diffusionjax-0.0.8-py3-none-any.whl (15 KB)
0.0.7	2023-04-22
diffusionjax-0.0.7-py3-none-any.whl (15 KB)
0.0.6	2023-03-21
diffusionjax-0.0.6-py3-none-any.whl (14 KB)
0.0.5	2023-03-16
diffusionjax-0.0.5-py3-none-any.whl (14 KB)
0.0.4	2023-03-04
diffusionjax-0.0.4-py3-none-any.whl (14 KB)
0.0.3	2023-01-24
diffusionjax-0.0.3-py3-none-any.whl (11 KB)
0.0.2	2023-02-24
diffusionjax-0.0.2-py3-none-any.whl (11 KB)
0.0.1	2023-01-24
diffusionjax-0.0.1-py3-none-any.whl (11 KB)
0.0.0	2023-01-24
diffusionjax-0.0.0-py3-none-any.whl (11 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue