PyPI page
Home page
Author:
None
Summary:
Implementation of Transformer Flows (Apple ML) in JAX and Equinox.
Latest version:
0.0.13
Required dependencies:
beartype
|
einops
|
equinox
|
jax
|
jaxtyping
|
matplotlib
|
ml-collections
|
optax
|
tqdm
Optional dependencies:
datasets
Downloads last day:
74
Downloads last week:
341
Downloads last month:
346