1. task_seq.py - This is the main file, which needs to be called with appropriate parameters for training a model.

2. The three .ipynb files compute the Gaussian robustness, projected out matrices for rich and lazy regime, using checkpoints obtained via running the task_seq.py file.