List of all items
Structs
- activation::PReLU
- batch_norm::BatchNorm
- batch_norm::BatchNormConfig
- conv::Conv1d
- conv::Conv1dConfig
- conv::Conv2d
- conv::Conv2dConfig
- conv::ConvTranspose1d
- conv::ConvTranspose1dConfig
- conv::ConvTranspose2d
- conv::ConvTranspose2dConfig
- embedding::Embedding
- func::Func
- func::FuncT
- group_norm::GroupNorm
- layer_norm::LayerNorm
- layer_norm::LayerNormConfig
- layer_norm::RmsNorm
- linear::Linear
- ops::Dropout
- optim::AdamW
- optim::ParamsAdamW
- optim::SGD
- rnn::GRU
- rnn::GRUConfig
- rnn::GRUState
- rnn::LSTM
- rnn::LSTMConfig
- rnn::LSTMState
- sequential::Sequential
- var_builder::Rename
- var_builder::Shard
- var_builder::ShardedSafeTensors
- var_builder::VarBuilderArgs
- var_map::VarMap
Enums
Traits
- Module
- ModuleT
- optim::Optimizer
- rnn::RNN
- var_builder::Backend
- var_builder::Renamer
- var_builder::SimpleBackend
Functions
- activation::prelu
- batch_norm::batch_norm
- conv::conv1d
- conv::conv1d_no_bias
- conv::conv2d
- conv::conv2d_no_bias
- conv::conv_transpose1d
- conv::conv_transpose1d_no_bias
- conv::conv_transpose2d
- conv::conv_transpose2d_no_bias
- embedding::embedding
- encoding::one_hot
- func::func
- func::func_t
- group_norm::group_norm
- layer_norm::layer_norm
- layer_norm::rms_norm
- linear::linear
- linear::linear_b
- linear::linear_no_bias
- loss::binary_cross_entropy_with_logit
- loss::cross_entropy
- loss::mse
- loss::nll
- ops::dropout
- ops::hard_sigmoid
- ops::leaky_relu
- ops::log_softmax
- ops::pixel_shuffle
- ops::pixel_unshuffle
- ops::replication_pad2d
- ops::rms_norm
- ops::rms_norm_slow
- ops::sigmoid
- ops::silu
- ops::softmax
- ops::softmax_last_dim
- ops::swiglu
- rnn::gru
- rnn::lstm
- rotary_emb::rope
- rotary_emb::rope_i
- rotary_emb::rope_i_slow
- rotary_emb::rope_slow
- rotary_emb::rope_thd
- sequential::seq