PyPI page
Home page
Author:
None
Summary:
REAX: A simple training framework for JAX-based projects
Latest version:
0.6.10
Required dependencies:
beartype
|
clu
|
einops
|
equinox
|
flax
|
fsspec
|
jax
|
jaxlib
|
jaxtyping
|
lightning-utilities
|
optax
|
pytray
|
stevedore
|
tensorboardx
|
tqdm
|
typing-extensions
Optional dependencies:
black
|
cloudpickle
|
docutils
|
flit
|
ipython
|
jupyter
|
lightning
|
mlflow
|
myst-nb
|
pandas
|
pandoc
|
pillow
|
pre-commit
|
pydata_sphinx_theme
|
pylint
|
pytest
|
pytest-cov
|
pytest-env
|
scikit-learn
|
sphinx
|
sphinx-autobuild
|
torch
Downloads last day:
2
Downloads last week:
65
Downloads last month:
338