lamp-core
lamp-core
lamp.autograd
Add
ArcTan
ArgMax
Assign
Autograd
AvgPool2D
BatchNorm
BatchNorm2D
BatchedMatMul
BinaryCrossEntropyWithLogitsLoss
CappedShiftedNegativeExponential
CastToPrecision
Cholesky
CholeskySolve
Concatenate
ConstAdd
ConstMult
Constant
Constant
ConstantWithGrad
ConstantWithoutGrad
Convolution
Cos
Cross
Debug
Diag
Div
Dropout
ElementWiseMaximum
ElementWiseMinimum
Embedding
EqWhere
EuclideanDistance
Exp
Expand
ExpandAs
Flatten
Gelu
GraphMemoryAllocationReport
HardSwish
IndexAdd
IndexAddToTarget
IndexFill
IndexSelect
Inv
L1Loss
LayerNormOp
LeakyRelu
Log
Log1p
LogDet
LogSoftMax
MaskFill
MaskSelect
MatMul
MaxPool1D
MaxPool2D
Mean
Mean
Minus
MseLoss
Mult
NllLoss
NoReduction
Norm2
OneHot
Op
PInv
Pow
PowConst
Reduction
Relu
RepeatInterleave
Reshape
ScatterAdd
Select
Sigmoid
Sin
Slice
Softplus
SparseFromValueAndIndex
SquaredFrobeniusMatrixNorm
Stack
Sum
Sum
Tan
Tanh
ToDense
Transpose
Variable
Variable
VariableNonConstant
VariableNonConstant
Variance
View
WeightNorm
Where
lamp.nn
AdamW
AdamW
AdversarialTraining
Attention
AttentionDecoder
BatchNorm
BatchNorm
Bias
RunningMean
RunningVar
Weights
BatchNorm2D
BatchNorm2D
Bias
Weights
Conv1D
Conv1D
Bias
Weights
Conv2D
Conv2D
Bias
Weights
Conv2DTransposed
Conv2DTransposed
Bias
Weights
Debug
Debug
DependentHyperparameter
Dropout
Dropout
EitherModule
EitherModule
Tag
Embedding
Embedding
Weights
FreeRunningRNN
FreeRunningRNN
Fun
Fun
GRU
GRU
BiasH
BiasR
BiasZ
WeightHh
WeightHr
WeightHz
WeightXh
WeightXr
WeightXz
GenericFun
GenericFun
GenericModule
GenericModule
InitState
InitState
InitStateSyntax
LSTM
LSTM
BiasC
BiasF
BiasI
BiasO
WeightHc
WeightHf
WeightHi
WeightHo
WeightXc
WeightXf
WeightXi
WeightXo
LayerNorm
LayerNorm
Bias
Scale
LeafTag
LearningRateSchedule
LearningRateSchedule
ReduceLROnPlateauState
LiftedModule
LiftedModule
Linear
Linear
Bias
Weights
Load
Load
LoadSyntax
LossCalculation
LossFunction
LossFunctions
BCEWithLogits
Identity
L1Loss
MSE
NLL
SequenceNLL
MLP
ActivationFunction
Gelu
HardSwish
NormType
BatchNorm
LayerNorm
NoNorm
NormType
Relu
Sigmoid
Swish1
MappedState
MappedState
ModelWithOptimizer
MultiheadAttention
MultiheadAttention
WeightsK
WeightsO
WeightsQ
WeightsV
NoTag
Optimizer
OptimizerHyperparameter
PTag
PTag
PerturbedLossCalculation
PositionalEmbedding
RAdam
RAdam
RNN
RNN
BiasH
WeightHh
WeightXh
Recursive
Recursive
ResidualModule
ResidualModule
SGDW
SGDW
Seq2
Seq2
Seq2Seq
Seq2Seq
Seq2SeqWithAttention
Seq2SeqWithAttention
Seq3
Seq3
Seq4
Seq4
Seq5
Seq5
Seq6
Seq6
SeqLinear
SeqLinear
Bias
Weight
Sequential
Sequential
Tag
Shampoo
Shampoo
SimpleLossCalculation
StatefulSeq2
StatefulSeq2
StatefulSeq3
StatefulSeq3
StatefulSeq4
StatefulSeq4
StatefulSeq5
StatefulSeq5
SupervisedModel
ToLift
ToMappedState
ToUnlift
ToWithInit
TrainingMode
TrainingMode
TrainingModeSyntax
TransformerEmbedding
TransformerEmbedding
Embedding
TransformerEncoder
TransformerEncoder
TransformerEncoderBlock
TransformerEncoderBlock
Bias1
Bias2
Weights1
Weights2
UnliftedModule
UnliftedModule
WeightNormLinear
WeightNormLinear
Bias
WeightsG
WeightsV
WithInit
WithInit
Yogi
Yogi
sequence
simple
statefulSequence
lamp.nn.bert
BertEncoder
BertEncoder
PositionalEmbeddingWeight
BertLoss
BertLoss
BertLossInput
BertLossInput
BertPretrainInput
BertPretrainInput
BertPretrainModule
BertPretrainModule
BertPretrainOutput
MaskedLanguageModelModule
MaskedLanguageModelModule
lamp.nn.graph
GCN
GCN
Graph
GraphAttention
GraphAttention
Weights
MPNN
MPNN
VertexPooling
Mean
PoolType
Sum
lamp.util
syntax
lamp-core
/
lamp.nn
/
ToLift
ToLift
implicit
class
ToLift
[
M
<:
Module
](
mod:
M
&
Module
)
Graph
Supertypes
class
Object
trait
Matchable
class
Any
Value members
Value members
Concrete methods
def
lift
:
LiftedModule
[
M
&
Module
]