Mila
Deep Neural Network Library
|
Files | |
Encoder.ixx | |
Implementation of the Encoder module for token and positional embeddings in transformer models. | |
EncoderConfig.ixx | |
Configuration interface for the Encoder module in the Mila DNN framework. | |
Linear.ixx | |
Implementation of the Linear (fully connected) module for neural networks. | |
LinearConfig.ixx | |
Configuration interface for the Linear module in the Mila DNN framework. | |
MultiHeadAttention.ixx | |
Implementation of multi-head attention mechanism for transformer architectures. | |
MultiHeadAttentionConfig.ixx | |
Configuration interface for the MultiHeadAttention module in the Mila DNN framework. | |
Residual.ixx | |
Implementation of the Residual connection module for neural networks. | |
ResidualConfig.ixx | |
Configuration interface for the Residual module in the Mila DNN framework. | |
Softmax.ixx | |
Implementation of the Softmax activation function module. | |
SoftmaxConfig.ixx | |
Configuration interface for the Softmax module in the Mila DNN framework. | |