PyPI page
Author:
Martin Uhrin
Summary:
REAX: A simple training framework for JAX-based projects
Latest version:
0.6.14
Required dependencies:
beartype
|
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:
4
Downloads last week:
74
Downloads last month:
973
7-day smoothing