|
Mila
Deep Neural Network Library
|
Mila public API umbrella module - the single supported entry point (import Mila;). More...
#include <string>#include <memory>#include <format>#include <exception>import Data.BpePreTokenizationMode;import Data.SpecialTokens;import Data.CharVocabularyConfig;import Data.CharTrainer;import Data.CharTokenizer;import Distribution.HuggingFaceHub;import Distribution.HttpClient;import Data.TokenSequenceLoader;import Distribution.ModelResolver;import Distribution.ModelHub;import Distribution.ModelPackage;import Distribution.Environment;import Serialization.PretrainedReader;import Serialization.SafeTensors;import Serialization.ArchiveSerializer;import Serialization.Metadata;import Data.Tokenizer;import Data.TokenizerType;import Dnn.Optimizers.AdamWConfig;import Dnn.Models.GemmaModel;import Dnn.Models.LlamaModel;import Dnn.Components.GemmaProtocol;import Dnn.Components.GemmaBlock;import Dnn.Components.IDecoderLayer;import Dnn.Components.LlamaTransformer;import Dnn.Components.GptTransformer;import Serialization.Mode;import Dnn.Components.GptBlock;import Dnn.Components.GatedMLP;import Dnn.Components.Residual;import Dnn.Quantization.KvCache.Policy;import Serialization.ModelArchive;import Dnn.Quantization.Weight.Policies;import Compute.DeviceRegistryHelpers;import Dnn.Components.GemmaConfig;import Compute.DeviceRegistrar;import Data.BpeTrainer;import Compute.CpuDevice;import Distribution.ModelManifest;import Compute.MemoryResource;import Logging.ConsoleSink;import Data.CharVocabulary;import Compute.DeviceTypeTraits.Cpu;import Data.BpeVocabulary;import Dnn.ComponentType;import Dnn.ComponentConfig;import Dnn.SamplingParams;import Serialization.Tensor;import Mila.Version;import Data.BpeTokenizer;import Dnn.Tensor;import Dnn.ActivationType;import Dnn.Models.GemmaModelConfig;import Compute.CpuMemoryResource;import Dnn.Models.LlamaModelConfig;import Dnn.Models.GptModel;import Dnn.CompositeComponent;import Compute.Device;import Dnn.ApproximationMethod;import Logging.NullSink;import Dnn.Component;import Serialization.OpenMode;import Logging.Logger;import Distribution.HttpTransport;import Serialization.ZipSerializer;import Compute.OptimizerBase;import Distribution.HttpTransportBackend;import Dnn.Components.LayerNorm;import Dnn.Components.Softmax;import Dnn.Components.Linear;import Compute.OperationBase;import Dnn.GenerateParams;import Dnn.Components.MLP;import Compute.IExecutionContext;import Dnn.Components.Rope;import Compute.ExecutionContextFactory;import Distribution.ModelCoordinate;import Compute.DeviceType;import Data.DataLoader;import Dnn.Components.GemmaTransformer;import Dnn.Network;import Compute.DeviceId;import Dnn.Model;import Compute.DeviceRegistry;import Dnn.LanguageModel;import Compute.DeviceTypeTraits;import Dnn.LanguageNetwork;import Dnn.NetworkFactory;import Dnn.GenerateStatus;import Dnn.RuntimeMode;import Dnn.ITensor;import Dnn.Components.Lpe;import Dnn.TensorBuffer;import Dnn.ModelType;import Dnn.TensorDataTypeTraits;import Dnn.Components.Gqa;import Distribution.Sha256;import Dnn.TensorTypes;import Compute.OperationType;import Dnn.TensorDataType;import Dnn.Optimizers.AdamW;import Dnn.TensorDataTypeMap;import Dnn.TensorHostTypeMap;import Dnn.Components.Swiglu;import Dnn.TensorOps;import Data.BpeVocabularyConfig;import Distribution.ModelStore;import Dnn.ConnectionType;import Core.RandomGenerator;import Logging.FileSink;import Dnn.Components.MultiHeadAttention;import Compute.OperationTraits;import Dnn.Components.Gelu;import Dnn.Components.Activation;import Dnn.Components.RmsNorm;import Dnn.Components.TokenEmbedding;Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
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. | |