PyPI page
Home page
Author:
Nick DePalma
License:
MIT
Summary:
Lightning-like framework for JAX
Latest version:
0.1.1
Required dependencies:
altair
|
datasets
|
dltype
|
flax
|
jax
|
jaxlib
|
numba
|
numpy
|
optax
|
orbax-checkpoint
|
progress-table
|
pydantic
|
pydantic-yaml
|
tensorboard
|
tensorboardx
|
wandb
Downloads last day:
1
Downloads last week:
4
Downloads last month:
12
7-day smoothing