PyPI page
Home page
Author:
Benjamin Boys and Jakiw Pidstrigach
License:
MIT
Summary:
diffusionjax is a simple and accessible diffusion models package in JAX
Latest version:
0.1.8
Required dependencies:
absl-py
|
flax
|
matplotlib
|
ml_collections
|
numpy
|
scipy
|
tqdm
|
wandb
Optional dependencies:
coveralls
|
flake8
|
jax
|
jaxlib
|
jaxtyping
|
mlkernels
|
mypy
|
optax
|
orbax-checkpoint
|
pre-commit
|
pylint
|
pytest
|
pytest-cov
|
pytest-xdist
|
ruff
|
setuptools_scm
|
setuptools_scm_git_archive
|
torch
|
typing-extensions
Downloads last day:
0
Downloads last week:
13
Downloads last month:
365