jaxnn

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: Foundation models in JAX/Flax
Latest version: 0.1.22
Required dependencies: flax | huggingface-hub | jax | numpy | orbax-checkpoint | pillow | torch | torchvision
Optional dependencies: pytest | timm | torch | torchvision

Downloads last day: 15
Downloads last week: 33
Downloads last month: 584