|
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.TokenSequenceLoader;import Data.BpePreTokenizationMode;import Data.BpeTokenizer;import Data.BpeVocabularyConfig;import Dnn.Components.LayerNorm;import Dnn.Network;import Dnn.GenerateParams;import Dnn.Components.Swiglu;import Dnn.Components.AttentionOutputGate;import Dnn.Components.Softmax;import Dnn.Components.MultiHeadAttention;import Dnn.ComponentType;import Dnn.ApproximationMethod;import Data.Tokenizer;import Dnn.Components.Gelu;import Dnn.TensorDataTypeMap;import Dnn.TensorDataTypeTraits;import Dnn.Components.QwenDeltaNetBlock;import Dnn.TensorTypes;import Dnn.Components.Lpe;import Dnn.TensorBuffer;import Dnn.Components.Gqa;import Dnn.Tensor;import Dnn.GenerateStatus;import Distribution.HttpTransportBackend;import Dnn.Model;import Compute.DeviceType;import Dnn.Components.ITransformerBlock;import Dnn.Components.QwenConfig;import Compute.CpuMemoryResource;import Dnn.Components.Router;import Dnn.Models.GptModel;import Compute.DeviceId;import Compute.ExecutionContextFactory;import Dnn.Models.LlamaModelConfig;import Compute.IExecutionContext;import Compute.GqaWorkspace;import Dnn.SamplingParams;import Dnn.Components.GptBlock;import Distribution.HuggingFaceHub;import Compute.DeviceRegistrar;import Dnn.LanguageModelNetwork;import Dnn.Components.QwenPrecisionPlan;import Dnn.TensorOps;import Dnn.ConnectionType;import Compute.OperationBase;import Logging.Logger;import Dnn.ActivationType;import Dnn.Components.Linear;import Dnn.ComponentConfig;import Dnn.Components.GptTransformer;import Dnn.Components.GemmaConfig;import Distribution.HttpTransport;import Logging.FileSink;import Compute.DeviceRegistryHelpers;import Logging.NullSink;import Data.CharVocabularyConfig;import Logging.ConsoleSink;import Dnn.NetworkFactory;import Dnn.LanguageModel;import Dnn.ModelType;import Compute.DeviceTypeTraits;import Distribution.HttpClient;import Distribution.ModelHub;import Compute.DeviceTypeTraits.Cpu;import Distribution.ModelStore;import Dnn.ITensor;import Compute.DeviceRegistry;import Dnn.Components.Activation;import Compute.MemoryResource;import Data.CharTokenizer;import Dnn.Component;import Distribution.ModelCoordinate;import Dnn.Components.RmsNorm;import Dnn.Components.LlamaTransformer;import Dnn.Components.TokenEmbedding;import Compute.OperationTraits;import Dnn.Components.Residual;import Dnn.Components.CausalConv1d;import Dnn.Models.GemmaProtocol;import Compute.CpuDevice;import Dnn.Quantization.Weight.PrecisionPlan;import Dnn.Components.GatedDeltaRule;import Dnn.Components.MLP;import Dnn.TensorDataType;import Compute.Observation;import Dnn.Components.MixtureOfExperts;import Dnn.Components.GemmaBlock;import Dnn.Components.GemmaTransformer;import Compute.OperationType;import Dnn.Components.QwenAttentionBlock;import Data.BpeVocabulary;import Dnn.Components.QwenTransformer;import Compute.OptimizerBase;import Dnn.Models.LlamaModel;import Dnn.Models.Conversation;import Dnn.Components.Rope;import Dnn.Models.GemmaModel;import Data.BpeTrainer;import Mila.Version;import Serialization.SafeTensors;import Data.SpecialTokens;import Dnn.Models.GemmaModelConfig;import Dnn.CompositeComponent;import Serialization.Tensor;import Distribution.ModelPackage;import Dnn.Models.QwenModel;import Core.RandomGenerator;import Dnn.Models.QwenModelConfig;import Distribution.Environment;import Dnn.Models.QwenProtocol;import Dnn.TensorHostTypeMap;import Compute.Device;import Serialization.ZipSerializer;import Dnn.Optimizers.AdamW;import Dnn.Quantization.KvCache.Policy;import Dnn.Optimizers.AdamWConfig;import Data.DataLoader;import Dnn.Components.GatedMLP;import Serialization.ModelArchive;import Data.TokenizerType;import Serialization.Mode;import Serialization.OpenMode;import Dnn.Quantization.Weight.Policies;import Serialization.Metadata;import Distribution.ModelManifest;import Dnn.RuntimeMode;import Serialization.ArchiveSerializer;import Serialization.WeightsReader;import Distribution.Sha256;import Distribution.ModelResolver;import Data.CharTrainer;import Data.CharVocabulary;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. | |