Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
LanguageModelNetwork.ixx File Reference

Abstract base for language model networks. More...

#include <cstdint>
#include <stdexcept>
#include <string>
import Dnn.TensorDataTypeTraits;
import Dnn.TensorDataType;
import Compute.DeviceTypeTraits;
import Compute.DeviceType;
import Dnn.TensorTypes;
import Dnn.Tensor;
import Dnn.Network;

Classes

class  Mila::Dnn::LanguageModelNetwork< TDeviceType, TPrecision >
 The type erasure boundary between a language model and its transformer. More...
struct  Mila::Dnn::SequenceLogLikelihood
 Teacher-forced log-likelihood of a token sequence under a model. More...

Namespaces

namespace  Mila
 Mila main API namespace.

Detailed Description

Abstract base for language model networks.