PyPI page
Home page
Author:
None
Summary:
Just Layers over JAX
Latest version:
0.0.8
Required dependencies:
einops
|
grain
|
humanize
|
jax
|
optax
|
orbax-checkpoint
|
plum-dispatch
|
pydantic
Optional dependencies:
jax
|
tpu-info
Downloads last day:
0
Downloads last week:
0
Downloads last month:
19