PyPI page
Home page
Author:
Erfan Zare Chavoshi
License:
Apache-2.0
Summary:
A JAX-only neural-network library with a PyTorch-shaped eager surface and Flax-NNX-style graph/state underneath.
Latest version:
0.0.1
Required dependencies:
jax
|
jaxlib
|
numpy
|
treescope
Optional dependencies:
jax
|
myst-nb
|
optax
|
pytest
|
ruff
|
sphinx
|
sphinx-autodoc-typehints
|
sphinx-book-theme
|
sphinx-design
Downloads last day:
4
Downloads last week:
10
Downloads last month:
131