Code for the paper "The Probabilistic Fault Tolerance of Neural Networks in the Continuous Limit", ICML 2020