PyPI page
Summary:
Implementations of popular vision models in Jax (Flax NNX)
Latest version:
0.0.7
Required dependencies:
einops
|
flax
|
jax
|
py7zr
|
urllib3
Optional dependencies:
jax-cuda12-plugin
|
jaxlib
|
torch
|
torchvision
Downloads last day:
1
Downloads last week:
8
Downloads last month:
22
7-day smoothing