This repository contains the reference implementation of AugMix, a data augmentation technique for improving robustness and uncertainty calibration in image classification models. It provides training scripts for CIFAR-10/100 and ImageNet using PyTorch, along with evaluation on their corrupted versions (CIFAR-C, ImageNet-C). The core augmentation logic is implemented in NumPy and PIL, and the training code applies a Jensen-Shannon Divergence consistency loss between augmented and clean images. The codebase is research-oriented, with scripts that are well-structured but heavily tied to the specific experimental setup of the paper. Dependencies are pinned but severely outdated (torch==1.2.0), making reproduction on modern systems challenging without adaptation.