Code for visualizing the loss landscape of neural nets - View it on GitHub