PyPI page
Home page
Author:
Shinapri
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTI...
Summary:
Build, train, and scale neural networks with JAX.
Latest version:
0.0.1
Required dependencies:
grain
|
jax
|
optax
|
orbax-checkpoint
|
qwix
Optional dependencies:
jax
|
tensorboardx
|
wandb
Downloads last day:
4
Downloads last week:
6
Downloads last month:
267
7-day smoothing