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