sgd_noise Environment conda env create -f environment.yml Features to add: [] saves plots to images [] add visualization for data at beginning [] images path