PyPI page
Home page
Author:
Erfan Zare Chavoshi
License:
AGPL-3.0-or-later
Summary:
SpectraX: a JAX-only neural-network library with a True MPMD pipeline parallelism for JAX — with an eager module API.
Latest version:
0.1.1
Required dependencies:
fsspec
|
jax
|
jaxlib
|
numpy
|
protobuf
|
tensorstore
|
treescope
Optional dependencies:
basedpyright
|
gcsfs
|
jax
|
myst-nb
|
optax
|
pytest
|
ruff
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-book-theme
|
sphinx-design
Downloads last day:
172
Downloads last week:
1,120
Downloads last month:
1,418