Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila Module Reference

Exported Modules

module  Dnn.Components.TokenEmbedding
module  Dnn.Components.Activation
module  Dnn.Components.MultiHeadAttention
module  Dnn.TensorHostTypeMap
module  Compute.OperationType
module  Dnn.Components.Gqa
module  Dnn.ConnectionType
module  Dnn.TensorDataTypeTraits
module  Dnn.TensorOps
module  Distribution.Sha256
module  Dnn.ModelType
module  Dnn.GenerateStatus
module  Core.RandomGenerator
module  Dnn.NetworkFactory
module  Distribution.ModelStore
module  Compute.DeviceRegistry
module  Dnn.Model
module  Data.DataLoader
module  Compute.DeviceType
module  Distribution.ModelCoordinate
module  Dnn.Components.Rope
module  Compute.IExecutionContext
module  Dnn.Components.MLP
module  Dnn.GenerateParams
module  Logging.FileSink
module  Dnn.Components.Softmax
module  Dnn.Components.LayerNorm
module  Distribution.HttpTransportBackend
module  Compute.OptimizerBase
module  Serialization.ZipSerializer
module  Distribution.HttpTransport
module  Logging.Logger
module  Dnn.Component
module  Compute.ExecutionContextFactory
module  Logging.NullSink
module  Compute.DeviceRegistrar
module  Dnn.Components.GptBlock
module  Dnn.ITensor
module  Dnn.Components.GptTransformer
module  Compute.DeviceTypeTraits
module  Logging.ConsoleSink
module  Dnn.Components.GemmaProtocol
module  Distribution.Environment
module  Dnn.Components.Linear
module  Dnn.Models.LlamaModel
module  Dnn.Network
module  Dnn.Components.IDecoderLayer
module  Dnn.Optimizers.AdamWConfig
module  Dnn.TensorDataTypeMap
module  Data.TokenizerType
module  Compute.DeviceTypeTraits.Cpu
module  Data.Tokenizer
module  Serialization.SafeTensors
module  Dnn.ComponentType
module  Data.BpePreTokenizationMode
module  Data.BpeVocabularyConfig
module  Dnn.RuntimeMode
module  Dnn.LanguageNetwork
module  Serialization.OpenMode
module  Compute.DeviceRegistryHelpers
module  Dnn.Models.GptModel
module  Serialization.Mode
module  Dnn.TensorDataType
module  Compute.OperationBase
module  Dnn.Models.GemmaModelConfig
module  Compute.DeviceId
module  Dnn.Components.GatedMLP
module  Data.CharTokenizer
module  Dnn.Optimizers.AdamW
module  Dnn.Models.GemmaModel
module  Data.CharVocabularyConfig
module  Dnn.TensorTypes
module  Compute.CpuMemoryResource
module  Distribution.HttpClient
module  Distribution.ModelPackage
module  Data.TokenSequenceLoader
module  Serialization.ArchiveSerializer
module  Data.CharTrainer
module  Dnn.ActivationType
module  Dnn.ApproximationMethod
module  Distribution.ModelResolver
module  Distribution.ModelHub
module  Dnn.Quantization.KvCache.Policy
module  Dnn.Components.RmsNorm
module  Dnn.CompositeComponent
module  Dnn.Components.LlamaTransformer
module  Dnn.Quantization.Weight.Policies
module  Serialization.PretrainedReader
module  Dnn.ComponentConfig
module  Dnn.Components.GemmaBlock
module  Dnn.Components.GemmaTransformer
module  Data.BpeVocabulary
module  Dnn.Components.Gelu
module  Dnn.TensorBuffer
module  Dnn.Components.Residual
module  Distribution.HuggingFaceHub
module  Serialization.ModelArchive
module  Dnn.Components.GemmaConfig
module  Dnn.Components.Swiglu
module  Compute.Device
module  Compute.CpuDevice
module  Mila.Version
module  Data.BpeTrainer
module  Distribution.ModelManifest
module  Data.SpecialTokens
module  Dnn.SamplingParams
module  Compute.MemoryResource
module  Data.BpeTokenizer
module  Data.CharVocabulary
module  Compute.OperationTraits
module  Serialization.Tensor
module  Dnn.Components.Lpe
module  Dnn.LanguageModel
module  Dnn.Tensor
module  Dnn.Models.LlamaModelConfig

Functions

bool Mila::initialize (unsigned int randomSeed=0, std::shared_ptr< Logging::Logger > sink=nullptr)
 Initializes the Mila framework.
void Mila::shutdown ()
 Shuts down the Mila framework and releases all resources.

Files

file  Mila/Src/Mila.ixx
 Mila public API umbrella module - the single supported entry point (import Mila;).