PyPI page
Home page
Author:
None
License:
Apache License
Version 2.0, January 2004
http://www.apache.org/licenses/
TERMS AND CONDITIONS FOR USE, REPRODUCTI...
Summary:
Elegant easy-to-use neural networks in JAX.
Latest version:
0.13.2
Required dependencies:
jax
|
jaxtyping
|
typing-extensions
|
wadler-lindig
Optional dependencies:
beartype
|
griffe
|
hippogriffe
|
jaxlib
|
mkdocs
|
mkdocs-include-exclude-files
|
mkdocs-ipynb
|
mkdocs-material
|
mkdocstrings
|
mkdocstrings-python
|
optax
|
pre-commit
|
pymdown-extensions
|
pytest
Downloads last day:
37,668
Downloads last week:
219,237
Downloads last month:
1,030,248