PyPI page
Home page
Author:
None
Summary:
A toolbox for mixed precision training via JAX.
Latest version:
0.1.8
Required dependencies:
equinox
|
jax
|
jaxtyping
|
optax
|
typing-extensions
|
wadler-lindig
Downloads last day:
0
Downloads last week:
22
Downloads last month:
40