PyPI page
Home page
Author:
Benjamin Bolte
Summary:
A library for fast Jax experimentation
Latest version:
0.4.4
Required dependencies:
attrs
|
chex
|
dpshdl
|
equinox
|
gitpython
|
importlib-resources
|
jax
|
jaxtyping
|
omegaconf
|
optax
|
orbax-checkpoint
|
pillow
|
psutil
|
requests
|
tensorboard
Optional dependencies:
black
|
darglint
|
mypy
|
pytest
|
ruff
|
types-pillow
|
types-psutil
|
types-requests
|
wandb
Downloads last day:
0
Downloads last week:
49
Downloads last month:
934