PyPI page
Home page
Author:
None
License:
MIT License
Copyright (c) [2025] [Jed Homer]
Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated docu...
Summary:
Score-based Diffusion models in JAX.
Latest version:
0.0.38
Required dependencies:
beartype
|
cloudpickle
|
diffrax
|
einops
|
einx
|
equinox
|
jax
|
matplotlib
|
ml-collections
|
numpy
|
optax
|
powerbox
|
scikit-learn
|
torch
|
torchvision
|
tqdm
Optional dependencies:
datasets
|
matplotlib
|
numpy
Downloads last day:
3
Downloads last week:
15
Downloads last month:
471