Last released Sep 29, 2024
An implementation of PSGD optimizer in JAX.
Last released Sep 28, 2024
None
Last released Aug 15, 2024
Run image classification experiments in JAX with ViT, resnet, cifar10, cifar100, imagenette, and imagenet.
Supported by