Functions — github.com/JonasGeiping/cramming

CodeHub MCPcopy Create free account

hub / github.com/JonasGeiping/cramming / functions

Functions411 in github.com/JonasGeiping/cramming

⨍ Functions 411◇ Types & classes 86

(self, closure=None)

cramming/backend/optimizers/optimizer_modifiers.py:35

(self, closure=None)

cramming/backend/optimizers/optimizer_modifiers.py:73

Performs a single optimization step. Args: closure (callable, optional): A closure that reevaluates the model and

cramming/backend/optimizers/shampoo/shampoo.py:289

Functiontokenize_function

cramming/data/pretraining_preparation.py:226

Methodtorch_call

Simplified call assuming all dicts in the list of examples have the same layout and contain tensors. Assume further that all these tensors con

cramming/backend/utils.py:202

Methodupdate_preconditioners

(self, grad: Tensor)

cramming/backend/optimizers/shampoo/shampoo_utils.py:122

Methodupdate_preconditioners

(self, grad: Tensor)

cramming/backend/optimizers/shampoo/shampoo_utils.py:190

Methodupdate_preconditioners

(self, grad: Tensor)

cramming/backend/optimizers/shampoo/shampoo_utils.py:330

Methodupdate_preconditioners

(self, grad: Tensor)

cramming/backend/optimizers/shampoo/shampoo_utils.py:574

Methodupdate_preconditioners

(self, grad: Tensor)

cramming/backend/optimizers/shampoo/shampoo_utils.py:670

Functionwandb_log

cramming/utils.py:455

← previous401–411 of 411, ranked by callers