Here is a list of all documented class members 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

~

- s -

• SafeTensorsWriter() : Mila::Dnn::Serialization::SafeTensorsWriter
• sample() : Mila::Dnn::Sampler< TDeviceType, TPrecision >, Mila::Dnn::TokenSampler< TDeviceType, TPrecision >
• sampleNext() : Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
• save() : Mila::Data::BpeVocabulary, Mila::Data::CharVocabulary, Mila::Data::TokenizerVocabulary, Mila::Dnn::Network< TDeviceType, TPrecision >
• save_() : Mila::Dnn::Gelu< TDeviceType, TPrecision >, Mila::Dnn::GemmaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::GptTransformer< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::LlamaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::Network< TDeviceType, TPrecision >, Mila::Dnn::Residual< TDeviceType, TPrecision >, Mila::Dnn::Softmax< TDeviceType, TPrecision >
• saveCheckpoint() : Mila::Dnn::GptModel< TDeviceType, TPrecision >
• saveFlatTensors() : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::LayerNorm< TDeviceType, TPrecision >, Mila::Dnn::Linear< TDeviceType, TComputePrecision, TWeightQuant >, Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision >, Mila::Dnn::RmsNorm< TDeviceType, TPrecision >, Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization >
• saveParameterToArchive() : Mila::Dnn::Component< TDeviceType, TPrecision >
• saveParameterToWriter() : Mila::Dnn::Component< TDeviceType, TPrecision >
• savePretrained() : Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
• seedSampler() : Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
• set() : Mila::Data::SerializationMetadata, Mila::Dnn::Optimizers::SerializationMetadata, Mila::Dnn::Serialization::SerializationMetadata, Mila::Dnn::SerializationMetadata
• setCurrentBatch() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
• setDefaultLogger() : Mila::Logging::Logger
• setEval() : Mila::Dnn::Model< TDeviceType, TPrecision >
• setExecutionContext() : Mila::Dnn::Component< TDeviceType, TPrecision >
• setGradients() : Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
• setIncludeSourceLocation() : Mila::Logging::ConsoleSink, Mila::Logging::FileSink
• setIncludeTimestamp() : Mila::Logging::ConsoleSink, Mila::Logging::FileSink
• setLearningRate() : Mila::Dnn::Optimizer< TDeviceType, TPrecision >, Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
• setLevel() : Mila::Logging::ConsoleSink, Mila::Logging::FileSink, Mila::Logging::Logger, Mila::Logging::NullSink
• setMetadata() : Mila::Dnn::Serialization::SafeTensorsWriter
• setName() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• setParameters() : Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
• setSeed() : Mila::Core::RandomGenerator
• setStageProbe() : Mila::Dnn::GemmaTransformer< TDeviceType, TPrecision, TWeightQuantization, TKvCachePolicy >, Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >
• setState() : Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::IDecoderLayer< TDeviceType, TPrecision >, Mila::Dnn::LlamaBlock< TDeviceType, TPrecision, TWeightQuant, TKvPolicy >
• setTrainingMode() : Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::Compute::Operation< TDeviceType, TComputePrecision >
• setUseFlashDecode() : Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >
• setUseFlashPrefill() : Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >
• shape() : Mila::Dnn::ITensor, Mila::Dnn::Tensor< TDataType, TMemoryResource >
• shouldInitializeParameters() : Mila::Dnn::BuildContext
• size() : Mila::Data::SerializationMetadata, Mila::Dnn::ITensor, Mila::Dnn::Optimizers::SerializationMetadata, Mila::Dnn::Serialization::SerializationMetadata, Mila::Dnn::SerializationMetadata, Mila::Dnn::Tensor< TDataType, TMemoryResource >, Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• size_in_bytes : Mila::Dnn::TensorDataTypeTraits< TensorDataType::BF16 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP16 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP32 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP4_E2M1 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP4_E3M0 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP8_E4M3 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::FP8_E5M2 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::INT16 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::INT32 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::INT8 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::UINT16 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::UINT32 >, Mila::Dnn::TensorDataTypeTraits< TensorDataType::UINT8 >
• Softmax() : Mila::Dnn::Softmax< TDeviceType, TPrecision >
• SoftmaxCrossEntropy() : Mila::Dnn::SoftmaxCrossEntropy< TDeviceType, TPrecision >
• source_metadata_ : Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
• spirv_bit_width() : Mila::Dnn::VulkanTensorTraits
• StageProbe : Mila::Dnn::LanguageNetwork< TDeviceType, TPrecision >
• standard() : Mila::Data::SpecialTokens
• start() : Mila::Dnn::Compute::CudaTimer
• step() : Mila::Dnn::Optimizer< TDeviceType, TPrecision >, Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
• stop() : Mila::Dnn::Compute::CudaTimer
• storageBytes() : Mila::Dnn::TensorBuffer< TDataType, TMemoryResource, TrackMemory >
• streamTensorBlobs() : Mila::Dnn::Serialization::PretrainedModelReader
• strides() : Mila::Dnn::Tensor< TDataType, TMemoryResource >
• supports() : Mila::Dnn::CpuTensorDataTypeTraits, Mila::Dnn::VulkanTensorTraits
• supportsComputeShaders() : Mila::Dnn::Compute::MetalDevice
• supportsGPUFamily() : Mila::Dnn::Compute::MetalDevice
• supportsKVCache() : Mila::Dnn::GemmaBlock< TDeviceType, TPrecision, kGlobal, TWeightQuant, TKvPolicy >, Mila::Dnn::GptBlock< TDeviceType, TPrecision >, Mila::Dnn::GroupedQueryAttention< TDeviceType, TComputePrecision, TKvPolicy >, Mila::Dnn::IDecoderLayer< TDeviceType, TPrecision >, Mila::Dnn::MultiHeadAttention< TDeviceType, TPrecision >
• supportsMixedPrecision() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
• synchronize() : Mila::Dnn::Activation< TDeviceType, TPrecision, TFn >, Mila::Dnn::Component< TDeviceType, TPrecision >, Mila::Dnn::CompositeComponent< TDeviceType, TPrecision >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cpu >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Cuda >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Metal >, Mila::Dnn::Compute::ExecutionContext< DeviceType::Vulkan >, Mila::Dnn::Compute::IExecutionContext, 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::Network< 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 >