PyPI 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:
0
Downloads last week:
18
Downloads last month:
11
7-day smoothing