List of all items

Structs

• array::array::Array
• autograd::AutogradContext
• autograd::ComputeNode
• autograd::NoGrad
• autograd::NodeId
• autograd::nn::BatchNorm1d
• autograd::nn::Conv1d
• autograd::nn::Dropout
• autograd::nn::Flatten
• autograd::nn::Linear
• autograd::nn::ReLU
• autograd::nn::Sequential
• autograd::nn::Sigmoid
• autograd::nn::Softmax
• autograd::optim::adabound::AdaBound
• autograd::optim::adadelta::AdaDelta
• autograd::optim::adagrad::AdaGrad
• autograd::optim::adam::Adam
• autograd::optim::adamw::AdamW
• autograd::optim::lamb::LAMB
• autograd::optim::lbfgs::LBFGS
• autograd::optim::lookahead::Lookahead
• autograd::optim::nadam::NAdam
• autograd::optim::radam::RAdam
• autograd::optim::rmsprop::RMSprop
• autograd::optim::rprop::Rprop
• autograd::optim::schedulers::CosineAnnealingLR
• autograd::optim::schedulers::ExponentialLR
• autograd::optim::schedulers::LinearWarmup
• autograd::optim::schedulers::ReduceLROnPlateau
• autograd::optim::schedulers::StepLR
• autograd::optim::sgd::SGD
• autograd::tensor::Tensor
• autograd::train::CrossEntropyLoss
• autograd::train::Dataset
• autograd::train::MSELoss
• autograd::train::Metrics
• autograd::train::Trainer
• autograd::train::TrainerBuilder
• backend::blas::BlasBackend
• backend::capabilities::BackendCapabilities
• backend::cpu::CpuBackend
• backend::cuda::CudaBackend
• backend::dispatch::BackendValidation
• backend::dispatch::DispatchTable
• backend::dispatch::RuntimeCapabilities
• backend::metal::MetalBackend
• backend::microbench::AdaptiveLookupTable
• backend::microbench::BenchConfig
• backend::microbench::SizeRange
• backend::webgpu::WebGpuBackend
• ir::graph::IRGraph
• ir::hlo::HloNode
• llo::LLOProgram
• llo::matmul::MatMul
• llo::model::ModelMetadata
• llo::model::OnnxGraph
• llo::model::OnnxModel
• llo::model::OnnxNode
• llo::model::OnnxTensor
• llo::model::TrainingState
• llo::reduction::Reduction

Enums

• array::dtype::DType
• array::dyn_array::DynArray
• array_view::ArrayView
• autograd::OpKind
• backend::SelectedBackend
• ir::hlo::HloOp
• llo::LloOp
• llo::elementwise::ElementwiseBinaryKind
• llo::elementwise::ElementwiseKind
• llo::elementwise::ElementwiseStrategy
• llo::elementwise::ElementwiseUnaryKind
• llo::linalg::LinalgKind
• llo::linalg::LinalgStrategy
• llo::model::LossKind
• llo::model::ModelKind
• llo::model::OnnxAttribute
• llo::model::OptimizerKind
• llo::model::TrainingKind
• llo::random::RandomKind
• llo::reduction::ReductionKind
• llo::reduction::ReductionStrategy
• llo::shape::ShapeKind
• llo::stats::StatsKind

Traits

• array::dtype::DTypeValue
• autograd::nn::Module
• autograd::optim::base::Optimizer
• autograd::optim::schedulers::Scheduler
• autograd::train::LossFunction

Macros

• no_grad

Functions

• array::promotion::cast_array
• array::promotion::promote_arrays
• array::promotion::promoted_dtype
• array::promotion::validate_binary_op
• autograd::backward::add_backward
• autograd::backward::batchnorm_backward
• autograd::backward::conv1d_backward
• autograd::backward::cos_backward
• autograd::backward::cross_entropy_backward
• autograd::backward::div_backward
• autograd::backward::dropout_backward
• autograd::backward::exp_backward
• autograd::backward::flatten_backward
• autograd::backward::log_backward
• autograd::backward::matmul_backward
• autograd::backward::mean_backward
• autograd::backward::mse_backward
• autograd::backward::mul_backward
• autograd::backward::pow_backward
• autograd::backward::relu_backward
• autograd::backward::reshape_backward
• autograd::backward::sigmoid_backward
• autograd::backward::sin_backward
• autograd::backward::softmax_backward
• autograd::backward::sqrt_backward
• autograd::backward::sub_backward
• autograd::backward::sum_backward
• autograd::backward::tan_backward
• autograd::backward::tanh_backward
• autograd::backward::transpose_backward
• autograd::is_grad_enabled
• autograd::set_grad_enabled
• backend::blas::dot_blas
• backend::blas::matmul_blas
• backend::blas::matmul_blas_parallel
• backend::cpu::batchnorm::batch_norm_1d_inference
• backend::cpu::batchnorm::batch_norm_1d_training
• backend::cpu::batchnorm::batchnorm_backward_naive
• backend::cpu::conv::conv1d_backward_naive
• backend::cpu::conv::conv1d_naive
• backend::cpu::dropout::dropout
• backend::cpu::matmul_scalar_direct
• backend::cpu::matmul_scalar_parallel
• backend::cpu::matmul_simd_direct
• backend::cpu::parallel::matmul_parallel
• backend::cpu::parallel::matmul_with_kernel
• backend::cpu::random::bernoulli_mask
• backend::cpu::random::rand_uniform
• backend::cpu::random::seed
• backend::cpu::scalar::dot_scalar
• backend::cpu::scalar::elementwise_scalar
• backend::cpu::scalar::reduce_last_axis_optimized
• backend::cpu::scalar::reduce_scalar
• backend::cpu::simd::dot_simd
• backend::cpu::simd::elementwise_simd
• backend::cpu::simd::elementwise_simd_supported
• backend::cpu::simd::matmul_simd
• backend::cpu::simd::reduce_simd
• backend::cpu::simd_conv::conv1d_simd
• backend::dispatch::dispatch_elementwise_generic
• backend::dispatch::get_backend_override
• backend::dispatch::get_dispatch_table
• backend::dispatch::init_dispatch_table
• backend::dispatch::kernel_elementwise_simd
• backend::dispatch::kernel_matmul_blas_direct
• backend::dispatch::kernel_matmul_metal
• backend::dispatch::kernel_matmul_scalar
• backend::dispatch::kernel_matmul_simd
• backend::dispatch::kernel_matmul_webgpu
• backend::dispatch::kernel_reduction_simd
• backend::dispatch::select_kernels
• backend::dispatch::set_backend_override
• backend::dispatch::validate_backends
• backend::heuristics::choose_elementwise_strategy
• backend::heuristics::choose_reduction_strategy
• backend::metal::elementwise_metal
• backend::metal::is_available_cached
• backend::metal::matmul_metal
• backend::metal::reduction_metal
• backend::microbench::benchmark_elementwise
• backend::microbench::benchmark_matmul
• backend::microbench::benchmark_reduction
• backend::webgpu::batchnorm::batch_norm_1d_inference_webgpu
• backend::webgpu::batchnorm::batch_norm_1d_training_webgpu
• backend::webgpu::broadcast_to_webgpu
• backend::webgpu::codegen::elementwise_wgsl
• backend::webgpu::codegen::reduction_wgsl
• backend::webgpu::conv::conv1d_webgpu
• backend::webgpu::dropout::dropout_webgpu
• backend::webgpu::elementwise_webgpu
• backend::webgpu::is_available_cached
• backend::webgpu::matmul_webgpu
• backend::webgpu::reduction_webgpu
• codegen::cpu::scalar_kernel_template
• codegen::cpu::simd_kernel_template
• codegen::cuda::kernel_template
• codegen::emit_program_for_backend
• codegen::metal::kernel_template
• codegen::webgpu::make_wgsl_for_elementwise
• ir::lowering::lower
• llo::array_to_onnx_tensor
• ops::batchnorm::batch_norm
• ops::conv::conv1d
• ops::dropout::dropout
• ops::elementwise::binary::add::add
• ops::elementwise::binary::div::div
• ops::elementwise::binary::mul::mul
• ops::elementwise::binary::pow::pow
• ops::elementwise::binary::sub::sub
• ops::elementwise::unary::abs::abs
• ops::elementwise::unary::acos::acos
• ops::elementwise::unary::asin::asin
• ops::elementwise::unary::atan::atan
• ops::elementwise::unary::cos::cos
• ops::elementwise::unary::exp::exp
• ops::elementwise::unary::leaky_relu::leaky_relu
• ops::elementwise::unary::log::log
• ops::elementwise::unary::neg::neg
• ops::elementwise::unary::relu::relu
• ops::elementwise::unary::sigmoid::sigmoid
• ops::elementwise::unary::sin::sin
• ops::elementwise::unary::softplus::softplus
• ops::elementwise::unary::sqrt::sqrt
• ops::elementwise::unary::tan::tan
• ops::elementwise::unary::tanh::tanh
• ops::export::export_to_json
• ops::export::export_to_onnx
• ops::linalg::dot::dot
• ops::linalg::matmul::matmul
• ops::model::array_to_onnx_tensor
• ops::model::create_add_node
• ops::model::create_linear_node
• ops::model::create_matmul_node
• ops::model::create_mlp
• ops::model::create_relu_node
• ops::model::create_softmax_node
• ops::model::deserialize_onnx
• ops::model::infer
• ops::model::load_checkpoint
• ops::model::load_onnx
• ops::model::save_checkpoint
• ops::model::save_onnx
• ops::model::serialize_onnx
• ops::reduction::argmax::argmax
• ops::reduction::max::max
• ops::reduction::mean::mean
• ops::reduction::min::min
• ops::reduction::sum::sum
• ops::reduction::variance::variance
• ops::shape::broadcast_to::broadcast_to
• ops::shape::broadcast_to::validate_broadcast_public
• ops::shape::concat::concat
• ops::shape::flatten::flatten
• ops::shape::reshape::reshape
• ops::shape::transpose::transpose
• ops::stats::cross_entropy::cross_entropy
• ops::stats::norm::norm
• ops::stats::softmax::softmax
• ops_inplace::dot_f32
• ops_inplace::elementwise_f32
• ops_inplace::elementwise_view
• ops_inplace::matmul_f32
• ops_inplace::reduce_f32
• startup::print_startup_log

Type Aliases

• autograd::BackwardFn
• backend::dispatch::DotFn
• backend::dispatch::ElementwiseFn
• backend::dispatch::MatmulFn
• backend::dispatch::ReductionFn

Constants

• backend::capabilities::BLAS_CAPABILITIES
• backend::capabilities::CUDA_CAPABILITIES
• backend::capabilities::METAL_CAPABILITIES
• backend::capabilities::SCALAR_CAPABILITIES
• backend::capabilities::SIMD_CAPABILITIES
• backend::capabilities::WEBGPU_CAPABILITIES
• backend::heuristics::ELEMENTWISE_GPU_MIN_ELEMENTS
• backend::heuristics::ELEMENTWISE_SIMD_MIN_ELEMENTS
• backend::heuristics::REDUCTION_PARALLEL_MIN