SYMBOL INDEX (10 symbols across 1 files) FILE: soap.py class SOAP (line 11) | class SOAP(optim.Optimizer): method __init__ (line 47) | def __init__( method merge_dims (line 79) | def merge_dims(self, grad, max_precond_dim): method step (line 109) | def step(self, closure = None): method init_preconditioner (line 214) | def init_preconditioner(self, grad, state, precondition_frequency=10, method project (line 240) | def project(self, grad, state, merge_dims=False, max_precond_dim=10000): method update_preconditioner (line 268) | def update_preconditioner(self, grad, state, method project_back (line 309) | def project_back(self, grad, state, merge_dims=False, max_precond_dim=... method get_orthogonal_matrix (line 337) | def get_orthogonal_matrix(self, mat): method get_orthogonal_matrix_QR (line 373) | def get_orthogonal_matrix_QR(self, state, max_precond_dim=10000, merge...