List of all items
Structs
- Array
- Device
- DtypeIter
- Stream
- StreamOrDevice
- error::Exception
- losses::BinaryCrossEntropy
- losses::BinaryCrossEntropyBuilder
- losses::CosineSimilarityLoss
- losses::CosineSimilarityLossBuilder
- losses::CrossEntropy
- losses::CrossEntropyBuilder
- losses::GaussianNllLoss
- losses::GaussianNllLossBuilder
- losses::HingeLoss
- losses::HingeLossBuilder
- losses::HuberLoss
- losses::HuberLossBuilder
- losses::KlDivLoss
- losses::KlDivLossBuilder
- losses::L1Loss
- losses::L1LossBuilder
- losses::LogCoshLoss
- losses::LogCoshLossBuilder
- losses::MarginRankingLoss
- losses::MarginRankingLossBuilder
- losses::MseLoss
- losses::MseLossBuilder
- losses::NllLoss
- losses::NllLossBuilder
- losses::SmoothL1Loss
- losses::SmoothL1LossBuilder
- losses::TripletLoss
- losses::TripletLossBuilder
- module::Param
- nested::NestedHashMap
- nn::Alibi
- nn::AlibiInput
- nn::AlibiInputBuilder
- nn::AvgPool1d
- nn::AvgPool2d
- nn::BatchNorm
- nn::BatchNormBuilder
- nn::Bilinear
- nn::BilinearBuilder
- nn::Celu
- nn::CeluBuilder
- nn::Conv1d
- nn::Conv1dBuilder
- nn::Conv2d
- nn::Conv2dBuilder
- nn::Conv3d
- nn::Conv3dBuilder
- nn::ConvTranspose1d
- nn::ConvTranspose1dBuilder
- nn::ConvTranspose2d
- nn::ConvTranspose2dBuilder
- nn::ConvTranspose3d
- nn::ConvTranspose3dBuilder
- nn::Dropout
- nn::Dropout2d
- nn::Dropout2dBuilder
- nn::Dropout3d
- nn::Dropout3dBuilder
- nn::DropoutBuilder
- nn::Embedding
- nn::Gelu
- nn::GeluBuilder
- nn::Glu
- nn::GluBuilder
- nn::GroupNorm
- nn::GroupNormBuilder
- nn::Gru
- nn::GruBuilder
- nn::HardSwish
- nn::InstanceNorm
- nn::InstanceNormBuilder
- nn::LayerNorm
- nn::LayerNormBuilder
- nn::LeakyRelu
- nn::LeakyReluBuilder
- nn::Linear
- nn::LinearBuilder
- nn::LogSigmoid
- nn::LogSoftmax
- nn::LogSoftmaxBuilder
- nn::Lstm
- nn::LstmBuilder
- nn::LstmInput
- nn::LstmInputBuilder
- nn::MaxPool1d
- nn::MaxPool2d
- nn::Mish
- nn::MultiHeadAttention
- nn::MultiHeadAttentionBuilder
- nn::MultiHeadAttentionInput
- nn::MultiHeadAttentionInputBuilder
- nn::Pool
- nn::Prelu
- nn::PreluBuilder
- nn::QuantizedEmbedding
- nn::QuantizedEmbeddingBuilder
- nn::QuantizedLinear
- nn::QuantizedLinearBuilder
- nn::Relu
- nn::Relu6
- nn::RmsNorm
- nn::RmsNormBuilder
- nn::Rnn
- nn::RnnBuilder
- nn::RnnInput
- nn::RnnInputBuilder
- nn::RopeInput
- nn::RopeInputBuilder
- nn::RotaryPositionalEncoding
- nn::RotaryPositionalEncodingBuilder
- nn::Selu
- nn::Sequential
- nn::Sigmoid
- nn::Silu
- nn::SinusoidalPositionalEncoding
- nn::SinusoidalPositionalEncodingBuilder
- nn::Softmax
- nn::SoftmaxBuilder
- nn::Softplus
- nn::Softsign
- nn::Step
- nn::StepBuilder
- nn::Tanh
- nn::Transformer
- nn::TransformerBuilder
- nn::TransformerInput
- nn::Upsample
- ops::indexing::Ellipsis
- ops::indexing::NewAxis
- ops::indexing::RangeIndex
- ops::indexing::StrideBy
- optimizers::AdaDelta
- optimizers::AdaDeltaBuilder
- optimizers::AdaGrad
- optimizers::AdaGradBuilder
- optimizers::Adafactor
- optimizers::AdafactorBuilder
- optimizers::AdafactorState
- optimizers::Adam
- optimizers::AdamBuilder
- optimizers::AdamW
- optimizers::AdamWBuilder
- optimizers::Adamax
- optimizers::AdamaxBuilder
- optimizers::Lion
- optimizers::LionBuilder
- optimizers::RmsProp
- optimizers::RmsPropBuilder
- optimizers::Sgd
- optimizers::SgdBuilder
- transforms::compile::Compiled
Enums
- DeviceType
- Dtype
- error::AdaDeltaBuildError
- error::AdafactorBuildError
- error::AsSliceError
- error::CrossEntropyBuildError
- error::DropoutBuildError
- error::IoError
- error::MultiHeadAttentionBuildError
- error::OptimizerStateLoadError
- error::RmsPropBuildError
- error::TransformerBulidError
- error::UnflattenError
- linalg::Ord
- losses::LossReduction
- nested::NestedValue
- nn::GeluApprox
- nn::UpsampleMode
- ops::PadMode
- ops::PadWidth
- ops::TensorDotDims
- ops::indexing::ArrayIndexOp
- quantization::MaybeQuantized
- random::ShapeOrCount
- utils::SingleOrPair
- utils::SingleOrTriple
- utils::SingleOrVec
Traits
- ArrayElement
- FromNested
- FromScalar
- builder::Buildable
- builder::Builder
- module::Module
- module::ModuleParameters
- module::ModuleParametersExt
- module::Parameter
- module::UnaryModule
- nn::Activation
- nn::IntoModuleValueAndGrad
- nn::Pooling
- nn::SequentialModuleItem
- ops::ClipBound
- ops::indexing::ArrayIndex
- ops::indexing::IndexMutOp
- ops::indexing::IndexOp
- ops::indexing::IntoStrideBy
- ops::indexing::TryIndexMutOp
- ops::indexing::TryIndexOp
- optimizers::Optimizer
- optimizers::OptimizerState
- quantization::Quantizable
- transforms::IntoGrad
- transforms::IntoKeyedValueAndGrad
- transforms::IntoValueAndGrad
- transforms::compile::CallMut
- transforms::compile::CallMutWithState
- transforms::compile::Compile
- transforms::compile::CompileWithState
- utils::IntoOption
- utils::ScalarOrArray
- utils::Updatable
Macros
Derive Macros
Functions
- fast::layer_norm
- fast::layer_norm_device
- fast::rms_norm
- fast::rms_norm_device
- fast::rope
- fast::rope_device
- fast::scaled_dot_product_attention
- fast::scaled_dot_product_attention_device
- fft::fft
- fft::fft2
- fft::fft2_device
- fft::fft_device
- fft::fftn
- fft::fftn_device
- fft::ifft
- fft::ifft2
- fft::ifft2_device
- fft::ifft_device
- fft::ifftn
- fft::ifftn_device
- fft::irfft
- fft::irfft2
- fft::irfft2_device
- fft::irfft_device
- fft::irfftn
- fft::irfftn_device
- fft::rfft
- fft::rfft2
- fft::rfft2_device
- fft::rfft_device
- fft::rfftn
- fft::rfftn_device
- linalg::cholesky
- linalg::cholesky_device
- linalg::cholesky_inv
- linalg::cholesky_inv_device
- linalg::cross
- linalg::cross_device
- linalg::eigh
- linalg::eigh_device
- linalg::eigvalsh
- linalg::eigvalsh_device
- linalg::inv
- linalg::inv_device
- linalg::norm
- linalg::norm_device
- linalg::norm_ord
- linalg::norm_ord_device
- linalg::norm_p
- linalg::norm_p_device
- linalg::pinv
- linalg::pinv_device
- linalg::qr
- linalg::qr_device
- linalg::svd
- linalg::svd_device
- linalg::tri_inv
- linalg::tri_inv_device
- module::update_parameters
- nn::build_quantized_linear
- nn::celu
- nn::elu
- nn::gelu
- nn::gelu_approximate
- nn::gelu_fast_approximate
- nn::glu
- nn::hard_swish
- nn::leaky_relu
- nn::log_sigmoid
- nn::log_softmax
- nn::mish
- nn::prelu
- nn::quantize
- nn::relu
- nn::relu6
- nn::selu
- nn::sigmoid
- nn::silu
- nn::softplus
- nn::softsign
- nn::step
- nn::value_and_grad
- ops::abs
- ops::abs_device
- ops::acos
- ops::acos_device
- ops::acosh
- ops::acosh_device
- ops::add
- ops::add_device
- ops::addmm
- ops::addmm_device
- ops::all
- ops::all_close
- ops::all_close_device
- ops::all_device
- ops::any
- ops::any_device
- ops::arange
- ops::arange_device
- ops::argpartition
- ops::argpartition_all
- ops::argpartition_all_device
- ops::argpartition_device
- ops::argsort
- ops::argsort_all
- ops::argsort_all_device
- ops::argsort_device
- ops::array_eq
- ops::array_eq_device
- ops::as_strided
- ops::as_strided_device
- ops::asin
- ops::asin_device
- ops::asinh
- ops::asinh_device
- ops::at_least_1d
- ops::at_least_1d_device
- ops::at_least_2d
- ops::at_least_2d_device
- ops::at_least_3d
- ops::at_least_3d_device
- ops::atan
- ops::atan_device
- ops::atanh
- ops::atanh_device
- ops::block_masked_mm
- ops::block_masked_mm_device
- ops::broadcast_arrays
- ops::broadcast_arrays_device
- ops::broadcast_to
- ops::broadcast_to_device
- ops::ceil
- ops::ceil_device
- ops::clip
- ops::clip_device
- ops::concatenate
- ops::concatenate_device
- ops::conv1d
- ops::conv1d_device
- ops::conv2d
- ops::conv2d_device
- ops::conv3d
- ops::conv3d_device
- ops::conv_general
- ops::conv_general_device
- ops::conv_transpose1d
- ops::conv_transpose1d_device
- ops::conv_transpose2d
- ops::conv_transpose2d_device
- ops::conv_transpose3d
- ops::conv_transpose3d_device
- ops::cos
- ops::cos_device
- ops::cosh
- ops::cosh_device
- ops::cummax
- ops::cummax_device
- ops::cummin
- ops::cummin_device
- ops::cumprod
- ops::cumprod_device
- ops::cumsum
- ops::cumsum_device
- ops::degrees
- ops::degrees_device
- ops::dequantize
- ops::dequantize_device
- ops::diag
- ops::diag_device
- ops::diagonal
- ops::diagonal_device
- ops::divide
- ops::divide_device
- ops::divmod
- ops::divmod_device
- ops::einsum
- ops::einsum_device
- ops::eq
- ops::eq_device
- ops::erf
- ops::erf_device
- ops::erfinv
- ops::erfinv_device
- ops::exp
- ops::exp_device
- ops::expand_dims
- ops::expand_dims_device
- ops::expm1
- ops::expm1_device
- ops::eye
- ops::eye_device
- ops::flatten
- ops::flatten_device
- ops::floor
- ops::floor_device
- ops::floor_divide
- ops::floor_divide_device
- ops::full
- ops::full_device
- ops::ge
- ops::ge_device
- ops::gt
- ops::gt_device
- ops::identity
- ops::identity_device
- ops::indexing::argmax
- ops::indexing::argmax_all
- ops::indexing::argmax_all_device
- ops::indexing::argmax_device
- ops::indexing::argmin
- ops::indexing::argmin_all
- ops::indexing::argmin_all_device
- ops::indexing::argmin_device
- ops::indexing::argpartition
- ops::indexing::argpartition_all
- ops::indexing::argpartition_all_device
- ops::indexing::argpartition_device
- ops::indexing::argsort
- ops::indexing::argsort_all
- ops::indexing::argsort_all_device
- ops::indexing::argsort_device
- ops::indexing::put_along_axis
- ops::indexing::put_along_axis_device
- ops::indexing::take
- ops::indexing::take_all
- ops::indexing::take_all_device
- ops::indexing::take_along_axis
- ops::indexing::take_along_axis_device
- ops::indexing::take_device
- ops::indexing::topk
- ops::indexing::topk_all
- ops::indexing::topk_all_device
- ops::indexing::topk_device
- ops::inner
- ops::inner_device
- ops::is_close
- ops::is_close_device
- ops::is_inf
- ops::is_inf_device
- ops::is_nan
- ops::is_nan_device
- ops::is_neg_inf
- ops::is_neg_inf_device
- ops::is_pos_inf
- ops::is_pos_inf_device
- ops::le
- ops::le_device
- ops::linspace
- ops::linspace_device
- ops::log
- ops::log10
- ops::log10_device
- ops::log1p
- ops::log1p_device
- ops::log2
- ops::log2_device
- ops::log_add_exp
- ops::log_add_exp_device
- ops::log_device
- ops::log_sum_exp
- ops::log_sum_exp_device
- ops::logical_and
- ops::logical_and_device
- ops::logical_not
- ops::logical_not_device
- ops::logical_or
- ops::logical_or_device
- ops::lt
- ops::lt_device
- ops::matmul
- ops::matmul_device
- ops::max
- ops::max_device
- ops::maximum
- ops::maximum_device
- ops::mean
- ops::mean_device
- ops::min
- ops::min_device
- ops::minimum
- ops::minimum_device
- ops::move_axis
- ops::move_axis_device
- ops::multiply
- ops::multiply_device
- ops::ne
- ops::ne_device
- ops::negative
- ops::negative_device
- ops::ones
- ops::ones_device
- ops::ones_dtype_device
- ops::ones_like
- ops::ones_like_device
- ops::outer
- ops::outer_device
- ops::pad
- ops::pad_device
- ops::partition
- ops::partition_all
- ops::partition_all_device
- ops::partition_device
- ops::power
- ops::power_device
- ops::prod
- ops::prod_device
- ops::quantize
- ops::quantize_device
- ops::quantized_matmul
- ops::quantized_matmul_device
- ops::radians
- ops::radians_device
- ops::reciprocal
- ops::reciprocal_device
- ops::remainder
- ops::remainder_device
- ops::repeat
- ops::repeat_all
- ops::repeat_all_device
- ops::repeat_device
- ops::reshape
- ops::reshape_device
- ops::round
- ops::round_device
- ops::rsqrt
- ops::rsqrt_device
- ops::sigmoid
- ops::sigmoid_device
- ops::sign
- ops::sign_device
- ops::sin
- ops::sin_device
- ops::sinh
- ops::sinh_device
- ops::softmax
- ops::softmax_all
- ops::softmax_all_device
- ops::softmax_device
- ops::sort
- ops::sort_all
- ops::sort_all_device
- ops::sort_device
- ops::split
- ops::split_device
- ops::split_equal
- ops::split_equal_device
- ops::sqrt
- ops::sqrt_device
- ops::square
- ops::square_device
- ops::squeeze
- ops::squeeze_device
- ops::stack
- ops::stack_all
- ops::stack_all_device
- ops::stack_device
- ops::std
- ops::std_device
- ops::subtract
- ops::subtract_device
- ops::sum
- ops::sum_device
- ops::swap_axes
- ops::swap_axes_device
- ops::tan
- ops::tan_device
- ops::tanh
- ops::tanh_device
- ops::tensordot
- ops::tensordot_device
- ops::tile
- ops::tile_device
- ops::transpose
- ops::transpose_all
- ops::transpose_all_device
- ops::transpose_device
- ops::tri
- ops::tri_device
- ops::variance
- ops::variance_device
- ops::where
- ops::where_device
- ops::which
- ops::which_device
- ops::zeros
- ops::zeros_device
- ops::zeros_dtype
- ops::zeros_dtype_device
- ops::zeros_like
- ops::zeros_like_device
- optimizers::clip_grad_norm
- random::bernoulli
- random::bernoulli_device
- random::categorical
- random::categorical_device
- random::gumbel
- random::gumbel_device
- random::key
- random::multivariate_normal
- random::multivariate_normal_device
- random::normal
- random::normal_device
- random::randint
- random::randint_device
- random::seed
- random::split
- random::split_device
- random::truncated_normal
- random::truncated_normal_device
- random::uniform
- random::uniform_device
- stop_gradient
- stop_gradient_device
- transforms::async_eval
- transforms::async_eval_params
- transforms::compile::clear_cache
- transforms::compile::compile
- transforms::compile::compile_with_state
- transforms::compile::disable_compile
- transforms::compile::enable_compile
- transforms::eval
- transforms::eval_params
- transforms::fallible_jvp
- transforms::fallible_vjp
- transforms::grad
- transforms::jvp
- transforms::keyed_value_and_grad
- transforms::value_and_grad
- transforms::vjp
Type Aliases
- complex64
- error::Result
- losses::CrossEntropyBuilderWeights
- module::FlattenedModuleParam
- module::FlattenedModuleParamMut
- module::FlattenedModuleParamRef
- module::ModuleParam
- module::ModuleParamMut
- module::ModuleParamRef
- nn::GruInput
- nn::GruInputBuilder
- nn::NonLinearity
- nn::Rope
- nn::RopeBuilder
- nn::Sinpe
- nn::SinpeBuilder
- optimizers::AdafactorBeta1
- optimizers::AdafactorBuilderBeta1
- optimizers::AdafactorBuilderLr
- optimizers::AdafactorEps
- optimizers::AdafactorLr
- optimizers::Betas
- optimizers::MaybeClippedGrads
- optimizers::State
- transforms::KeyedGrad
- transforms::KeyedParameters