PyPI page
Home page
Author:
None
Summary:
High-performance composable JAX library
Latest version:
0.3.1
Required dependencies:
equinox
|
grain
|
jax
|
jaxlib
|
numpy
|
optax
|
orbax-checkpoint
|
pytest-asyncio
Optional dependencies:
beartype
|
chex
|
complexipy
|
flatbuffers
|
import-linter
|
interrogate
|
jaxlint
|
jaxtyping
|
libcst
|
matplotlib
|
pandas
|
pytest
|
pytest-asyncio
|
pytest-benchmark
|
pytest-cov
|
ruff
|
seaborn
|
ty
|
tyro
Downloads last day:
107
Downloads last week:
291
Downloads last month:
521