FFCV: Fast Forward Computer Vision (and other ML workloads!)
A library for experimenting with, training and evaluating neural networks, with a focus on adversarial robustness.
Open on GitHub