Module laplace.curvature.curvlinops

Classes

class CurvlinopsInterface (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels')

Interface for Curvlinops backend. https://github.com/f-dangel/curvlinops

Ancestors

Subclasses

Inherited members

class CurvlinopsGGN (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels', stochastic: bool = False)

Implementation of the GGNInterface using Curvlinops.

Ancestors

Inherited members

class CurvlinopsEF (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels')

Implementation of EFInterface using Curvlinops.

Ancestors

Inherited members

class CurvlinopsHessian (model: nn.Module, likelihood: Likelihood | str, last_layer: bool = False, subnetwork_indices: torch.LongTensor | None = None, dict_key_x: str = 'input_ids', dict_key_y: str = 'labels')

Implementation of the full Hessian using Curvlinops.

Ancestors

Inherited members