PyPI page
Home page
Author:
None
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:
2
Downloads last week:
67
Downloads last month:
548