Curvlinops
Getting started
How to use Curvlinops
Curvlinops
Linear operators
Spectral density approximation
Trace approximation
Diagonal approximation
Frobenius norm approximation
Experimental
Code samples
Internals
Internals
Curvlinops
Index
Index
_
|
A
|
C
|
D
|
E
|
F
|
G
|
H
|
J
|
K
|
L
|
M
|
N
|
R
|
S
|
T
|
X
_
__init__() (curvlinops.CGInverseLinearOperator method)
(curvlinops.EFLinearOperator method)
(curvlinops.EKFACLinearOperator method)
(curvlinops.experimental.ActivationHessianLinearOperator method)
(curvlinops.FisherMCLinearOperator method)
(curvlinops.GGNLinearOperator method)
(curvlinops.HessianLinearOperator method)
(curvlinops.JacobianLinearOperator method)
(curvlinops.KFACInverseLinearOperator method)
(curvlinops.KFACLinearOperator method)
(curvlinops.LanczosApproximateSpectrumCached method)
(curvlinops.LSMRInverseLinearOperator method)
(curvlinops.NeumannInverseLinearOperator method)
(curvlinops.SubmatrixLinearOperator method)
(curvlinops.TransposedJacobianLinearOperator method)
_SUPPORTED_FISHER_TYPE (curvlinops.EKFACLinearOperator attribute)
(curvlinops.KFACLinearOperator attribute)
_SUPPORTED_KFAC_APPROX (curvlinops.KFACLinearOperator attribute)
_SUPPORTED_LOSSES (curvlinops.KFACLinearOperator attribute)
_SUPPORTED_MODULES (curvlinops.KFACLinearOperator attribute)
A
ActivationHessianLinearOperator (class in curvlinops.experimental)
approximate_spectrum() (curvlinops.LanczosApproximateSpectrumCached method)
C
CGInverseLinearOperator (class in curvlinops)
compute_eigenvalue_correction_linear_weight_sharing() (in module curvlinops.ekfac)
D
det (curvlinops.EKFACLinearOperator property)
(curvlinops.KFACLinearOperator property)
E
EFLinearOperator (class in curvlinops)
EKFACLinearOperator (class in curvlinops)
EMPIRICAL (curvlinops.FisherType attribute)
EXPAND (curvlinops.KFACType attribute)
F
FisherMCLinearOperator (class in curvlinops)
FisherType (class in curvlinops)
FIXED_DATA_ORDER (curvlinops.FisherMCLinearOperator attribute)
(curvlinops.JacobianLinearOperator attribute)
(curvlinops.TransposedJacobianLinearOperator attribute)
FORWARD_ONLY (curvlinops.FisherType attribute)
frobenius_norm (curvlinops.EKFACLinearOperator property)
(curvlinops.KFACLinearOperator property)
from_state_dict() (curvlinops.EKFACLinearOperator class method)
(curvlinops.KFACLinearOperator class method)
G
GGNLinearOperator (class in curvlinops)
H
HessianLinearOperator (class in curvlinops)
hutchinson_diag() (in module curvlinops)
hutchinson_squared_fro (class in curvlinops)
hutchinson_trace() (in module curvlinops)
hutchpp_trace() (in module curvlinops)
J
JacobianLinearOperator (class in curvlinops)
K
KFACInverseLinearOperator (class in curvlinops)
KFACLinearOperator (class in curvlinops)
KFACType (class in curvlinops)
L
lanczos_approximate_log_spectrum() (in module curvlinops)
lanczos_approximate_spectrum() (in module curvlinops)
LanczosApproximateSpectrumCached (class in curvlinops)
load_state_dict() (curvlinops.EKFACLinearOperator method)
(curvlinops.KFACLinearOperator method)
logdet (curvlinops.EKFACLinearOperator property)
(curvlinops.KFACLinearOperator property)
loss_hessian_matrix_sqrt() (in module curvlinops.kfac_utils)
LSMRInverseLinearOperator (class in curvlinops)
M
MC (curvlinops.FisherType attribute)
N
NeumannInverseLinearOperator (class in curvlinops)
R
REDUCE (curvlinops.KFACType attribute)
S
SELF_ADJOINT (curvlinops.EFLinearOperator attribute)
(curvlinops.experimental.ActivationHessianLinearOperator attribute)
(curvlinops.FisherMCLinearOperator attribute)
(curvlinops.GGNLinearOperator attribute)
(curvlinops.HessianLinearOperator attribute)
(curvlinops.KFACInverseLinearOperator attribute)
(curvlinops.KFACLinearOperator attribute)
set_submatrix() (curvlinops.SubmatrixLinearOperator method)
state_dict() (curvlinops.EKFACLinearOperator method)
(curvlinops.KFACLinearOperator method)
SubmatrixLinearOperator (class in curvlinops)
supported_losses (curvlinops.FisherMCLinearOperator attribute)
SUPPORTS_BLOCKS (curvlinops.HessianLinearOperator attribute)
T
trace (curvlinops.EKFACLinearOperator property)
(curvlinops.KFACLinearOperator property)
TransposedJacobianLinearOperator (class in curvlinops)
TYPE2 (curvlinops.FisherType attribute)
X
xdiag() (in module curvlinops)
xtrace() (in module curvlinops)