JAX fast adversarial training using FGSM
This is a JAX implementation of Fast adversarial training using FGSM on the CIFAR10 dataset.
Smooth adversarial training
This is a JAX implementation of Fast adversarial training using FGSM on the CIFAR10 dataset.
Smooth adversarial training