Here is a list of all documented functions with links to the class documentation for each member:

A

B

C

D

E

F

G

H

I

K

L

M

N

O

P

R

S

T

U

V

W

Z

~

- o -

• onBuilding() : Mila::Dnn::Activation< TDeviceType, TPrecision, TFn >, Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::GatedMLP< TDeviceType, TPrecision, TGate >, Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::GemmaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::MLP< TDeviceType, TPrecision >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >, Mila::Dnn::Residual< TDeviceType, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::Rope< TDeviceType, TPrecision >, Mila::Dnn::Softmax< TDeviceType, TPrecision >, Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision, TGate >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization >
• onExecutionContextSet() : Mila::Dnn::Activation< TDeviceType, TPrecision, TFn >, Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >, Mila::Dnn::Residual< TDeviceType, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::Rope< TDeviceType, TPrecision >, Mila::Dnn::Softmax< TDeviceType, TPrecision >, Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision, TGate >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization >
• onGenerating() : Mila::Dnn::GemmaModel< TDeviceType, TPrecision >, Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LanguageModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >
• onTraining() : Mila::Dnn::GemmaModel< TDeviceType, TPrecision >, Mila::Dnn::GptModel< TDeviceType, TPrecision >, Mila::Dnn::LlamaModel< TDeviceType, TPrecision >, Mila::Dnn::Model< TDeviceType, TPrecision >
• onTrainingModeChanging() : Mila::Dnn::Activation< TDeviceType, TPrecision, TFn >, Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, Mila::Dnn::GatedMLP< TDeviceType, TPrecision, TGate >, Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::GemmaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::MLP< TDeviceType, TPrecision >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >, Mila::Dnn::Residual< TDeviceType, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::Rope< TDeviceType, TPrecision >, Mila::Dnn::Softmax< TDeviceType, TPrecision >, Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TPrecision >, Mila::Dnn::Swiglu< TDeviceType, TPrecision, TGate >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization >
• open() : Mila::Distribution::ModelPackage, Mila::Dnn::Serialization::Serializer, Mila::Dnn::Serialization::ZipSerializer
• operator()() : std::hash< Mila::Dnn::Compute::DeviceId >
• operator+=() : Mila::Dnn::MemoryStats
• operator<=>() : Mila::Dnn::Compute::DeviceId
• operator=() : Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• operator==() : Mila::Dnn::TensorShape
• operator[]() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• optimize() : Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >
• origin() : Mila::Distribution::ModelRecord