PyPI page
Home page
Author:
Shaymukhametov Rinat
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.2.0
Required dependencies:
einops
|
flax
|
huggingface-hub
|
jax
|
numpy
|
orbax-checkpoint
|
pillow
|
torch
|
torchvision
Optional dependencies:
pytest
|
timm
|
torch
|
torchvision
Downloads last day:
1
Downloads last week:
1
Downloads last month:
14
7-day smoothing