Class Index

A

B

C

D

E

F

G

H

I

L

M

N

O

P

R

S

T

U

V

Z

A

Activation (Mila::Dnn)

ActivationConfig (Mila::Dnn)

AdamWConfig (Mila::Dnn::Optimizers)

AdamWOptimizer (Mila::Dnn::Optimizers)

ArchiveSerializer (Mila::Dnn::Serialization)

AxisPartition (Mila::Dnn)

B

GemmaBlock::BlockBuildContexts (Mila::Dnn)

LlamaBlock::BlockBuildContexts (Mila::Dnn)

BpeTokenizer (Mila::Data)

BpeTrainer (Mila::Data)

BpeVocabulary (Mila::Data)

BpeVocabularyConfig (Mila::Data)

BufferedTokenStreamer (Mila::Dnn)

BuildContext (Mila::Dnn)

C

CharTokenizer (Mila::Data)

CharTrainer (Mila::Data)

CharVocabulary (Mila::Data)

CharVocabularyConfig (Mila::Data)

Component (Mila::Dnn)

ComponentConfig (Mila::Dnn)

ComponentFactory (Mila::Dnn)

CompositeComponent (Mila::Dnn)

ConsoleSink (Mila::Logging)

ConstantLRScheduler (Mila::Dnn)

CosineLRScheduler (Mila::Dnn)

CpuDevice (Mila::Dnn::Compute)

CpuDeviceRegistrar (Mila::Dnn::Compute)

CpuMemoryResource (Mila::Dnn::Compute)

CpuTensorDataTypeTraits (Mila::Dnn)

CrossEntropyConfig (Mila::Dnn)

CudaDataTypeTraits (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::BF16 > (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::FP16 > (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::FP32 > (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::FP8_E4M3 > (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::FP8_E5M2 > (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::INT32 > (Mila::Dnn::Compute::Cuda)

CudaDataTypeTraits< TensorDataType::INT8 > (Mila::Dnn::Compute::Cuda)

CudaDevice (Mila::Dnn::Compute)

CudaDeviceMemoryResource (Mila::Dnn::Compute)

CudaDeviceProps (Mila::Dnn::Compute)

CudaDeviceRegistrar (Mila::Dnn::Compute)

CudaManagedMemoryResource (Mila::Dnn::Compute)

CudaPinnedMemoryResource (Mila::Dnn::Compute)

CudaTimer (Mila::Dnn::Compute)

CurlHttpTransport (Mila::Distribution)

D

DataLoader (Mila::Data)

Decoder (Mila::Dnn)

dependent_false (Mila::Dnn)

Device (Mila::Dnn::Compute)

DeviceAccessible (Mila::Dnn::Compute)

DeviceConstructionKey (Mila::Dnn::Compute)

DeviceId (Mila::Dnn::Compute)

DeviceMemoryInfo (Mila::Dnn::Compute)

DeviceRegistrar (Mila::Dnn::Compute)

DeviceRegistry (Mila::Dnn::Compute)

DeviceTypeTraits (Mila::Dnn::Compute)

DeviceTypeTraits< DeviceType::Cpu > (Mila::Dnn::Compute)

DeviceTypeTraits< DeviceType::Cuda > (Mila::Dnn::Compute)

E

ExecutionContext (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)

F

FetchReport (Mila::Distribution)

FileSink (Mila::Logging)

functor_of (Mila::Dnn)

functor_of< ActivationType::Gelu > (Mila::Dnn)

functor_of< ActivationType::LeakyRelu > (Mila::Dnn)

functor_of< ActivationType::Mish > (Mila::Dnn)

functor_of< ActivationType::None > (Mila::Dnn)

functor_of< ActivationType::Relu > (Mila::Dnn)

functor_of< ActivationType::Sigmoid > (Mila::Dnn)

functor_of< ActivationType::Silu > (Mila::Dnn)

functor_of< ActivationType::Tanh > (Mila::Dnn)

G

GatedMLP (Mila::Dnn)

GatedMLPConfig (Mila::Dnn)

Gelu (Mila::Dnn)

GeluConfig (Mila::Dnn)

GemmaBlock (Mila::Dnn)

GemmaBlockWorkspace (Mila::Dnn)

GemmaConfig (Mila::Dnn)

GemmaModel (Mila::Dnn)

GemmaModelConfig (Mila::Dnn)

GemmaToolCall (Mila::Dnn::Gemma)

GemmaTransformer (Mila::Dnn)

GenerateParams (Mila::Dnn)

GptBlock (Mila::Dnn)

GptBlockConfig (Mila::Dnn)

GptConfig (Mila::Dnn)

GptModel (Mila::Dnn)

GptModelConfig (Mila::Dnn)

GptTransformer (Mila::Dnn)

GqaConfig (Mila::Dnn)

GqaState (Mila::Dnn::Compute)

GroupedQueryAttention (Mila::Dnn)

H

hash< Mila::Dnn::Compute::DeviceId > (std)

HostAccessible (Mila::Dnn::Compute)

HttpClient (Mila::Distribution)

HttpFetch (Mila::Distribution)

HttpHeader (Mila::Distribution)

HttpRequest (Mila::Distribution)

HttpResponse (Mila::Distribution)

HttpResult (Mila::Distribution)

HubModel (Mila::Distribution)

HuggingFaceHub (Mila::Distribution)

I

IDecoderLayer (Mila::Dnn)

IExecutionContext (Mila::Dnn::Compute)

IHttpTransport (Mila::Distribution)

IKvCacheLifecycle (Mila::Dnn::Compute)

IKvInference (Mila::Dnn::Compute)

IModelHub (Mila::Distribution)

InstallOptions (Mila::Distribution)

IPackedKvInference (Mila::Dnn::Compute)

IPositionalDecode (Mila::Dnn::Compute)

IPositionalPairedOp (Mila::Dnn::Compute)

ITensor (Mila::Dnn)

ITensorBlob (Mila::Dnn::Serialization)

L

LanguageModel (Mila::Dnn)

LanguageModelConfig (Mila::Dnn)

LanguageNetwork (Mila::Dnn)

LayerNorm (Mila::Dnn)

LayerNormConfig (Mila::Dnn)

LearningRateScheduler (Mila::Dnn)

Linear (Mila::Dnn)

LinearConfig (Mila::Dnn)

LinearLRScheduler (Mila::Dnn)

LlamaBlock (Mila::Dnn)

LlamaConfig (Mila::Dnn)

LlamaModel (Mila::Dnn)

LlamaModelConfig (Mila::Dnn)

LlamaTransformer (Mila::Dnn)

Logger (Mila::Logging)

Loss (Mila::Dnn)

Lpe (Mila::Dnn)

LpeConfig (Mila::Dnn)

M

MemoryAllocationStats (Mila::Dnn::Compute)

MemoryResource (Mila::Dnn::Compute)

MemoryResourceTraits (Mila::Dnn::Compute)

MemoryResourceTraits< CpuMemoryResource > (Mila::Dnn::Compute)

MemoryResourceTraits< CudaDeviceMemoryResource > (Mila::Dnn::Compute)

MemoryResourceTraits< CudaManagedMemoryResource > (Mila::Dnn::Compute)

MemoryResourceTraits< CudaPinnedMemoryResource > (Mila::Dnn::Compute)

MemoryStats (Mila::Dnn)

MetalDevice (Mila::Dnn::Compute)

MetalDevicePlugin (Mila::Dnn::Compute)

MetalMemoryResource (Mila::Dnn::Compute)

MilaFileHeader (Mila::Data)

MLP (Mila::Dnn)

MLPConfig (Mila::Dnn)

Model (Mila::Dnn)

ModelArchive (Mila::Dnn::Serialization)

ModelConfig (Mila::Dnn)

ModelCoordinate (Mila::Distribution)

ModelFile (Mila::Distribution)

ModelManifest (Mila::Distribution)

ModelPackage (Mila::Distribution)

ModelRecord (Mila::Distribution)

ModelResolver (Mila::Distribution)

ModelStore (Mila::Distribution)

MultiAxisPartition (Mila::Dnn)

MultiHeadAttention (Mila::Dnn)

MultiHeadAttentionConfig (Mila::Dnn)

N

Network (Mila::Dnn)

NetworkFactory (Mila::Dnn)

NoKvCompression (Mila::Dnn::Quant::KvCache)

NoWeightQuant (Mila::Dnn::Quant::Weight)

NullHttpTransport (Mila::Distribution)

NullSink (Mila::Logging)

NvtxRange (Mila::Profiling)

O

Operation (Mila::Dnn::Compute)

OperationTraits (Mila::Dnn::Compute)

Optimizer (Mila::Dnn)

P

PackageRequest (Mila::Distribution)

PackageValidation (Mila::Distribution)

PerChannelFp8 (Mila::Dnn::Quant::Weight)

PerChannelKvFp8 (Mila::Dnn::Quant::KvCache)

PerGroupFp4 (Mila::Dnn::Quant::Weight)

PerGroupInt4 (Mila::Dnn::Quant::Weight)

PretrainedMetadata (Mila::Dnn::Serialization)

PretrainedModelReader (Mila::Dnn::Serialization)

PruneOptions (Mila::Distribution)

R

RandomGenerator (Mila::Core)

RemovalReport (Mila::Distribution)

Residual (Mila::Dnn)

ResidualConfig (Mila::Dnn)

RmsNorm (Mila::Dnn)

RmsNormConfig (Mila::Dnn)

Rope (Mila::Dnn)

RopeConfig (Mila::Dnn)

S

SafeTensorsWriter (Mila::Dnn::Serialization)

Sampler (Mila::Dnn)

SamplingConfig (Mila::Dnn)

SamplingParams (Mila::Dnn)

ModelArchive::ScopedScope (Mila::Dnn::Serialization)

SerializationMetadata (Mila::Data)

SerializationMetadata (Mila::Dnn::Optimizers)

SerializationMetadata (Mila::Dnn::Serialization)

SerializationMetadata (Mila::Dnn)

Serializer (Mila::Dnn::Serialization)

Sha256 (Mila::Distribution)

SlidingWindowKvCache (Mila::Dnn::Quant::KvCache)

Softmax (Mila::Dnn)

SoftmaxConfig (Mila::Dnn)

SoftmaxCrossEntropy (Mila::Dnn)

SpecialTokens (Mila::Data)

StepLogger (Mila::Utils)

StoredModel (Mila::Distribution)

StoreUsage (Mila::Distribution)

Swiglu (Mila::Dnn)

SwigluConfig (Mila::Dnn)

T

Tensor (Mila::Dnn)

TensorBlob (Mila::Dnn::Serialization)

TensorBlobMetadata (Mila::Dnn::Serialization)

TensorBlobView (Mila::Dnn::Serialization)

TensorBuffer (Mila::Dnn)

TensorDataTypeMap (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap (Mila::Dnn)

TensorDataTypeMap< __nv_fp8_e4m3 > (Mila::Dnn)

TensorDataTypeMap< __nv_fp8_e5m2 > (Mila::Dnn)

TensorDataTypeMap< float > (Mila::Dnn)

TensorDataTypeMap< half > (Mila::Dnn)

TensorDataTypeMap< nv_bfloat16 > (Mila::Dnn)

TensorDataTypeMap< std::int16_t > (Mila::Dnn)

TensorDataTypeMap< std::int32_t > (Mila::Dnn)

TensorDataTypeMap< std::int8_t > (Mila::Dnn)

TensorDataTypeMap< std::uint16_t > (Mila::Dnn)

TensorDataTypeMap< std::uint32_t > (Mila::Dnn)

TensorDataTypeMap< std::uint8_t > (Mila::Dnn)

TensorDataTypeMap< TensorDataType::BF16 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::FP16 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::FP32 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::FP4_E2M1 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::FP4_E3M0 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::FP8_E4M3 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::FP8_E5M2 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::INT16 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::INT32 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::INT8 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::UINT16 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::UINT32 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeMap< TensorDataType::UINT8 > (Mila::Dnn::Compute::Cuda)

TensorDataTypeTraits (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 > (Mila::Dnn)

TensorHostTypeMap (Mila::Dnn)

TensorHostTypeMap< TensorDataType::BF16 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::FP16 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::FP32 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::FP8_E4M3 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::FP8_E5M2 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::INT16 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::INT32 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::INT8 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::UINT16 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::UINT32 > (Mila::Dnn)

TensorHostTypeMap< TensorDataType::UINT8 > (Mila::Dnn)

TensorMetadata (Mila::Dnn::Serialization)

TensorOps (Mila::Dnn)

TensorShape (Mila::Dnn)

TokenEmbedding (Mila::Dnn)

TokenEmbeddingConfig (Mila::Dnn)

Tokenizer (Mila::Data)

TokenizerTrainer (Mila::Data)

TokenizerVocabulary (Mila::Data)

TokenSampler (Mila::Dnn)

TokenSequenceLoader (Mila::Data)

TokenSequenceLoaderConfig (Mila::Data)

TrackedMemoryResource (Mila::Dnn::Compute)

TrainerFactory (Mila::Data)

U

UniqueIdGenerator (Mila::Dnn)

V

Version (Mila)

VulkanDevice (Mila::Dnn::Compute)

VulkanMemoryResource (Mila::Dnn::Compute)

VulkanTensorTraits (Mila::Dnn)

Z

ZipSerializer (Mila::Dnn::Serialization)