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:
15
Downloads last week:
38
Downloads last month:
51