dmft_wide_networks

Code to solve DMFT equations for infinite width feature learning networks and reproduce the figures in our Neurips 2022 paper.

1. Nonlinear DMFT equations with Monte-Carlo solver
2. Deep Linear DMFT solver
3. Two Layer ReLU network experiments
4. Code to reproduce the CIFAR-10 experiment can be found in scratch_cnn_expt.ipynb