|
Mila 0.13.48
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.BpeTrainer;import Data.BpeVocabularyConfig;import Data.BpeVocabulary;import Data.CharVocabulary;import Data.CharTrainer;import Data.CharTokenizer;import Serialization.ZipSerializer;import Serialization.ArchiveSerializer;import Serialization.ModelArchive;import Serialization.Metadata;import Serialization.OpenMode;import Serialization.Mode;import Data.DataLoader;import Dnn.Optimizers.AdamWConfig;import Dnn.Optimizers.AdamW;import Dnn.Models.LlamaModel;import Dnn.Models.GptModel;import Dnn.Components.LlamaTransformer;import Dnn.Components.GptTransformer;import Data.TokenizerType;import Dnn.Components.GptBlock;import Dnn.Components.Softmax;import Dnn.Components.Linear;import Dnn.ComponentType;import Compute.OperationRegistryHelpers;import Core.RandomGenerator;import Data.Tokenizer;import Compute.OperationRegistry;import Data.SpecialTokens;import Compute.DeviceRegistrar;import Compute.DeviceTypeTraits.Cpu;import Dnn.Network;import Dnn.TensorInitializers;import Dnn.Tensor;import Dnn.Component;import Compute.DeviceTypeTraits;import Dnn.ConnectionType;import Compute.DeviceRegistryHelpers;import Dnn.LanguageNetwork;import Compute.DeviceType;import Compute.ExecutionContextFactory;import Compute.CpuDevice;import Compute.OptimizerBase;import Compute.DeviceId;import Dnn.ITensor;import Dnn.TensorHostTypeMap;import Logging.ConsoleSink;import Dnn.TensorDataTypeTraits;import Logging.FileSink;import Dnn.Models.LlamaModelConfig;import Dnn.TensorTypes;import Data.BpeTokenizer;import Dnn.Components.Residual;import Compute.OperationBase;import Compute.UnaryOperation;import Logging.NullSink;import Dnn.Components.RmsNorm;import Data.CharVocabularyConfig;import Compute.OperationType;import Compute.CpuMemoryResource;import Compute.Device;import Compute.IExecutionContext;import Compute.MemoryResource;import Dnn.LanguageModel;import Dnn.Components.Swiglu;import Data.TokenSequenceLoader;import Compute.OperationsRegistrar;import Dnn.ComponentConfig;import Dnn.CompositeComponent;import Dnn.Components.MultiHeadAttention;import Dnn.NetworkFactory;import Dnn.Model;import Dnn.TensorBuffer;import Dnn.TensorDataType;import Dnn.TensorDataTypeMap;import Compute.DeviceRegistry;import Dnn.TensorOps;import Dnn.ApproximationMethod;import Dnn.ActivationType;import Compute.BinaryOperation;import Dnn.RuntimeMode;import Dnn.Components.Gqa;import Mila.Version;import Dnn.Components.Lpe;import Dnn.Components.MLP;import Logging.Logger;import Dnn.Components.Gelu;import Dnn.Components.Rope;import Dnn.Components.LayerNorm;import Compute.OperationTraits;Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::detail |
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. | |
Variables | |
| std::shared_ptr< Logging::Logger > | Mila::detail::g_defaultLogger |