image-classification-jax

Run image classification experiments in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install image-classification-jax

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
0.1.4 2024-12-18    
0.1.3 2024-10-07    
0.1.2 2024-08-15    
0.1.1 2024-08-15    
0.1.0 2024-08-07    

Issues with this package?

Page last updated 2024-12-19 01:07:51 UTC