Closed aleximmer closed 4 months ago
Jacobians can also be computed naively but for general models using either pytorch-functional or using loops. This should be an option if the layers cannot be extended using either backpack or asdfghjkl.
pytorch-functional
Addressed in #146.
Jacobians can also be computed naively but for general models using either
pytorch-functional
or using loops. This should be an option if the layers cannot be extended using either backpack or asdfghjkl.