PyTorch original implementation of "Making Hard Problems Easier with Custom Data Distributions and Loss Regularization A Case Study in Modular Arithmetic" (ICML 2025). - View it on GitHub