PyPI page
Home page
Author:
Evan Walters
Summary:
Run image classification experiments in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet.
Latest version:
0.1.4
Required dependencies:
einops
|
flax
|
jax
|
numpy
|
optax
|
psgd-jax
|
tensorflow-cpu
|
tensorflow-datasets
|
wandb
Downloads last day:
7
Downloads last week:
408
Downloads last month:
436