GitHub repository: marcellodebernardi/loss-landscapes
Primary language: Python
Description: Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch