List of all items

Structs

• constrained_module::Constrained
• manifolds::Euclidean
• manifolds::matrix_groups::OrthogonalGroup
• manifolds::sphere::Sphere
• manifolds::steifiel::SteifielsManifold
• optimizers::ManifoldRGD
• optimizers::ManifoldRGDConfig
• optimizers::ManifoldRGDState
• optimizers::ManifoldRGDStateItem
• optimizers::RiemannianAdam
• optimizers::RiemannianAdamConfig
• optimizers::RiemannianAdamState
• optimizers::RiemannianAdamStateItem
• optimizers::multiple::MultiManifoldOptimizer
• optimizers::multiple::MultiManifoldOptimizerConfig
• prelude::Bool
• prelude::Float
• prelude::Int
• prelude::Shape
• prelude::Tensor
• prelude::TensorData
• prelude::nn::BatchNorm
• prelude::nn::BatchNormConfig
• prelude::nn::BatchNormRecord
• prelude::nn::BatchNormRecordItem
• prelude::nn::BiLstm
• prelude::nn::BiLstmConfig
• prelude::nn::BiLstmRecord
• prelude::nn::BiLstmRecordItem
• prelude::nn::Dropout
• prelude::nn::DropoutConfig
• prelude::nn::Embedding
• prelude::nn::EmbeddingConfig
• prelude::nn::EmbeddingRecord
• prelude::nn::EmbeddingRecordItem
• prelude::nn::GLU
• prelude::nn::GateController
• prelude::nn::GateControllerRecord
• prelude::nn::GateControllerRecordItem
• prelude::nn::Gelu
• prelude::nn::GroupNorm
• prelude::nn::GroupNormConfig
• prelude::nn::GroupNormRecord
• prelude::nn::GroupNormRecordItem
• prelude::nn::HardSigmoid
• prelude::nn::HardSigmoidConfig
• prelude::nn::InstanceNorm
• prelude::nn::InstanceNormConfig
• prelude::nn::InstanceNormRecord
• prelude::nn::InstanceNormRecordItem
• prelude::nn::LayerNorm
• prelude::nn::LayerNormConfig
• prelude::nn::LayerNormRecord
• prelude::nn::LayerNormRecordItem
• prelude::nn::LeakyRelu
• prelude::nn::LeakyReluConfig
• prelude::nn::Linear
• prelude::nn::LinearConfig
• prelude::nn::LinearRecord
• prelude::nn::LinearRecordItem
• prelude::nn::Lstm
• prelude::nn::LstmConfig
• prelude::nn::LstmRecord
• prelude::nn::LstmRecordItem
• prelude::nn::LstmState
• prelude::nn::PRelu
• prelude::nn::PReluConfig
• prelude::nn::PReluRecord
• prelude::nn::PReluRecordItem
• prelude::nn::PositionalEncoding
• prelude::nn::PositionalEncodingConfig
• prelude::nn::PositionalEncodingRecord
• prelude::nn::PositionalEncodingRecordItem
• prelude::nn::Relu
• prelude::nn::RmsNorm
• prelude::nn::RmsNormConfig
• prelude::nn::RmsNormRecord
• prelude::nn::RmsNormRecordItem
• prelude::nn::RotaryEncoding
• prelude::nn::RotaryEncodingConfig
• prelude::nn::RotaryEncodingRecord
• prelude::nn::RotaryEncodingRecordItem
• prelude::nn::Sigmoid
• prelude::nn::SwiGlu
• prelude::nn::SwiGluConfig
• prelude::nn::SwiGluRecord
• prelude::nn::SwiGluRecordItem
• prelude::nn::Tanh
• prelude::nn::Unfold4d
• prelude::nn::Unfold4dConfig
• prelude::nn::activation::GLU
• prelude::nn::activation::Gelu
• prelude::nn::activation::HardSigmoid
• prelude::nn::activation::HardSigmoidConfig
• prelude::nn::activation::LeakyRelu
• prelude::nn::activation::LeakyReluConfig
• prelude::nn::activation::PRelu
• prelude::nn::activation::PReluConfig
• prelude::nn::activation::PReluRecord
• prelude::nn::activation::PReluRecordItem
• prelude::nn::activation::Relu
• prelude::nn::activation::Sigmoid
• prelude::nn::activation::SwiGlu
• prelude::nn::activation::SwiGluConfig
• prelude::nn::activation::SwiGluRecord
• prelude::nn::activation::SwiGluRecordItem
• prelude::nn::activation::Tanh
• prelude::nn::loss::BinaryCrossEntropyLoss
• prelude::nn::loss::BinaryCrossEntropyLossConfig
• prelude::nn::loss::BinaryCrossEntropyLossRecord
• prelude::nn::loss::BinaryCrossEntropyLossRecordItem
• prelude::nn::loss::CosineEmbeddingLoss
• prelude::nn::loss::CosineEmbeddingLossConfig
• prelude::nn::loss::CrossEntropyLoss
• prelude::nn::loss::CrossEntropyLossConfig
• prelude::nn::loss::CrossEntropyLossRecord
• prelude::nn::loss::CrossEntropyLossRecordItem
• prelude::nn::loss::HuberLoss
• prelude::nn::loss::HuberLossConfig
• prelude::nn::loss::MseLoss
• prelude::nn::loss::PoissonNllLoss
• prelude::nn::loss::PoissonNllLossConfig
• prelude::nn::modules::BatchNorm
• prelude::nn::modules::BatchNormConfig
• prelude::nn::modules::BatchNormRecord
• prelude::nn::modules::BatchNormRecordItem
• prelude::nn::modules::BiLstm
• prelude::nn::modules::BiLstmConfig
• prelude::nn::modules::BiLstmRecord
• prelude::nn::modules::BiLstmRecordItem
• prelude::nn::modules::Dropout
• prelude::nn::modules::DropoutConfig
• prelude::nn::modules::Embedding
• prelude::nn::modules::EmbeddingConfig
• prelude::nn::modules::EmbeddingRecord
• prelude::nn::modules::EmbeddingRecordItem
• prelude::nn::modules::GateController
• prelude::nn::modules::GateControllerRecord
• prelude::nn::modules::GateControllerRecordItem
• prelude::nn::modules::GroupNorm
• prelude::nn::modules::GroupNormConfig
• prelude::nn::modules::GroupNormRecord
• prelude::nn::modules::GroupNormRecordItem
• prelude::nn::modules::InstanceNorm
• prelude::nn::modules::InstanceNormConfig
• prelude::nn::modules::InstanceNormRecord
• prelude::nn::modules::InstanceNormRecordItem
• prelude::nn::modules::LayerNorm
• prelude::nn::modules::LayerNormConfig
• prelude::nn::modules::LayerNormRecord
• prelude::nn::modules::LayerNormRecordItem
• prelude::nn::modules::Linear
• prelude::nn::modules::LinearConfig
• prelude::nn::modules::LinearRecord
• prelude::nn::modules::LinearRecordItem
• prelude::nn::modules::Lstm
• prelude::nn::modules::LstmConfig
• prelude::nn::modules::LstmRecord
• prelude::nn::modules::LstmRecordItem
• prelude::nn::modules::LstmState
• prelude::nn::modules::PositionalEncoding
• prelude::nn::modules::PositionalEncodingConfig
• prelude::nn::modules::PositionalEncodingRecord
• prelude::nn::modules::PositionalEncodingRecordItem
• prelude::nn::modules::RmsNorm
• prelude::nn::modules::RmsNormConfig
• prelude::nn::modules::RmsNormRecord
• prelude::nn::modules::RmsNormRecordItem
• prelude::nn::modules::RotaryEncoding
• prelude::nn::modules::RotaryEncodingConfig
• prelude::nn::modules::RotaryEncodingRecord
• prelude::nn::modules::RotaryEncodingRecordItem
• prelude::nn::modules::Unfold4d
• prelude::nn::modules::Unfold4dConfig
• prelude::nn::modules::attention::GeneratePaddingMask
• prelude::nn::modules::attention::MhaCache
• prelude::nn::modules::attention::MhaInput
• prelude::nn::modules::attention::MhaOutput
• prelude::nn::modules::attention::MultiHeadAttention
• prelude::nn::modules::attention::MultiHeadAttentionConfig
• prelude::nn::modules::attention::MultiHeadAttentionRecord
• prelude::nn::modules::attention::MultiHeadAttentionRecordItem
• prelude::nn::modules::cache::TensorCache
• prelude::nn::modules::conv::Conv1d
• prelude::nn::modules::conv::Conv1dConfig
• prelude::nn::modules::conv::Conv1dRecord
• prelude::nn::modules::conv::Conv1dRecordItem
• prelude::nn::modules::conv::Conv2d
• prelude::nn::modules::conv::Conv2dConfig
• prelude::nn::modules::conv::Conv2dRecord
• prelude::nn::modules::conv::Conv2dRecordItem
• prelude::nn::modules::conv::Conv3d
• prelude::nn::modules::conv::Conv3dConfig
• prelude::nn::modules::conv::Conv3dRecord
• prelude::nn::modules::conv::Conv3dRecordItem
• prelude::nn::modules::conv::ConvTranspose1d
• prelude::nn::modules::conv::ConvTranspose1dConfig
• prelude::nn::modules::conv::ConvTranspose1dRecord
• prelude::nn::modules::conv::ConvTranspose1dRecordItem
• prelude::nn::modules::conv::ConvTranspose2d
• prelude::nn::modules::conv::ConvTranspose2dConfig
• prelude::nn::modules::conv::ConvTranspose2dRecord
• prelude::nn::modules::conv::ConvTranspose2dRecordItem
• prelude::nn::modules::conv::ConvTranspose3d
• prelude::nn::modules::conv::ConvTranspose3dConfig
• prelude::nn::modules::conv::ConvTranspose3dRecord
• prelude::nn::modules::conv::ConvTranspose3dRecordItem
• prelude::nn::modules::conv::DeformConv2d
• prelude::nn::modules::conv::DeformConv2dConfig
• prelude::nn::modules::conv::DeformConv2dRecord
• prelude::nn::modules::conv::DeformConv2dRecordItem
• prelude::nn::modules::gru::Gru
• prelude::nn::modules::gru::GruConfig
• prelude::nn::modules::gru::GruRecord
• prelude::nn::modules::gru::GruRecordItem
• prelude::nn::modules::interpolate::Interpolate1d
• prelude::nn::modules::interpolate::Interpolate1dConfig
• prelude::nn::modules::interpolate::Interpolate2d
• prelude::nn::modules::interpolate::Interpolate2dConfig
• prelude::nn::modules::lstm::BiLstm
• prelude::nn::modules::lstm::BiLstmConfig
• prelude::nn::modules::lstm::BiLstmRecord
• prelude::nn::modules::lstm::BiLstmRecordItem
• prelude::nn::modules::lstm::Lstm
• prelude::nn::modules::lstm::LstmConfig
• prelude::nn::modules::lstm::LstmRecord
• prelude::nn::modules::lstm::LstmRecordItem
• prelude::nn::modules::lstm::LstmState
• prelude::nn::modules::norm::BatchNorm
• prelude::nn::modules::norm::BatchNormConfig
• prelude::nn::modules::norm::BatchNormRecord
• prelude::nn::modules::norm::BatchNormRecordItem
• prelude::nn::modules::norm::GroupNorm
• prelude::nn::modules::norm::GroupNormConfig
• prelude::nn::modules::norm::GroupNormRecord
• prelude::nn::modules::norm::GroupNormRecordItem
• prelude::nn::modules::norm::InstanceNorm
• prelude::nn::modules::norm::InstanceNormConfig
• prelude::nn::modules::norm::InstanceNormRecord
• prelude::nn::modules::norm::InstanceNormRecordItem
• prelude::nn::modules::norm::LayerNorm
• prelude::nn::modules::norm::LayerNormConfig
• prelude::nn::modules::norm::LayerNormRecord
• prelude::nn::modules::norm::LayerNormRecordItem
• prelude::nn::modules::norm::RmsNorm
• prelude::nn::modules::norm::RmsNormConfig
• prelude::nn::modules::norm::RmsNormRecord
• prelude::nn::modules::norm::RmsNormRecordItem
• prelude::nn::modules::pool::AdaptiveAvgPool1d
• prelude::nn::modules::pool::AdaptiveAvgPool1dConfig
• prelude::nn::modules::pool::AdaptiveAvgPool2d
• prelude::nn::modules::pool::AdaptiveAvgPool2dConfig
• prelude::nn::modules::pool::AvgPool1d
• prelude::nn::modules::pool::AvgPool1dConfig
• prelude::nn::modules::pool::AvgPool2d
• prelude::nn::modules::pool::AvgPool2dConfig
• prelude::nn::modules::pool::MaxPool1d
• prelude::nn::modules::pool::MaxPool1dConfig
• prelude::nn::modules::pool::MaxPool2d
• prelude::nn::modules::pool::MaxPool2dConfig
• prelude::nn::modules::transformer::PositionWiseFeedForward
• prelude::nn::modules::transformer::PositionWiseFeedForwardConfig
• prelude::nn::modules::transformer::PositionWiseFeedForwardRecord
• prelude::nn::modules::transformer::PositionWiseFeedForwardRecordItem
• prelude::nn::modules::transformer::TransformerDecoder
• prelude::nn::modules::transformer::TransformerDecoderAutoregressiveCache
• prelude::nn::modules::transformer::TransformerDecoderConfig
• prelude::nn::modules::transformer::TransformerDecoderInput
• prelude::nn::modules::transformer::TransformerDecoderLayer
• prelude::nn::modules::transformer::TransformerDecoderLayerRecord
• prelude::nn::modules::transformer::TransformerDecoderLayerRecordItem
• prelude::nn::modules::transformer::TransformerDecoderRecord
• prelude::nn::modules::transformer::TransformerDecoderRecordItem
• prelude::nn::modules::transformer::TransformerEncoder
• prelude::nn::modules::transformer::TransformerEncoderAutoregressiveCache
• prelude::nn::modules::transformer::TransformerEncoderConfig
• prelude::nn::modules::transformer::TransformerEncoderInput
• prelude::nn::modules::transformer::TransformerEncoderLayer
• prelude::nn::modules::transformer::TransformerEncoderLayerRecord
• prelude::nn::modules::transformer::TransformerEncoderLayerRecordItem
• prelude::nn::modules::transformer::TransformerEncoderRecord
• prelude::nn::modules::transformer::TransformerEncoderRecordItem

Enums

• prelude::nn::Initializer
• prelude::nn::LinearLayout
• prelude::nn::PaddingConfig1d
• prelude::nn::PaddingConfig2d
• prelude::nn::PaddingConfig3d
• prelude::nn::activation::Activation
• prelude::nn::activation::ActivationConfig
• prelude::nn::activation::ActivationRecord
• prelude::nn::activation::ActivationRecordItem
• prelude::nn::loss::Reduction
• prelude::nn::modules::LinearLayout
• prelude::nn::modules::attention::SeqLengthOption
• prelude::nn::modules::interpolate::InterpolateMode
• prelude::nn::modules::norm::Normalization
• prelude::nn::modules::norm::NormalizationConfig
• prelude::nn::modules::norm::NormalizationRecord
• prelude::nn::modules::norm::NormalizationRecordItem

Traits

• constrained_module::ConstrainedModule
• lie_group::MonoidManifold
• manifolds::Manifold
• optimizers::hessian_optimizer::LessSimpleHessianOptimizer
• optimizers::hessian_optimizer::SimpleHessianOptimizer
• optimizers::many_steps::LessSimpleOptimizer
• optimizers::multiple::ManifoldOptimizable
• prelude::Backend
• prelude::Config
• prelude::DeviceOps
• prelude::ElementConversion
• prelude::Module
• prelude::SliceArg
• prelude::ToElement

Macros

• prelude::s

Derive Macros

• prelude::Config
• prelude::Module

Functions

• prelude::nn::generate_sinusoids
• prelude::nn::modules::attention::generate_autoregressive_mask
• prelude::nn::modules::attention::generate_padding_mask
• prelude::nn::modules::generate_sinusoids

Type Aliases

• prelude::Device