List of all items
Structs
- attention::AttentionShape
- crypto::Sha256Ctx
- crypto::Sha512Ctx
- kv_cache::KvCacheView
- layers::DenseLayerDesc
- layers::LayerPlan
- metrics::RunningMean
- model_config::TransformerConfig
- model_format::DecodedCounts
- model_format::HeaderV1
- network::NetworkStats
- network::NeuralNetwork
- profiler::OpCounter
- rnn_api::RnnDenseCapacity
- rnn_format::BlobMeta
- rnn_format::RnnHandle
- runtime::BudgetFit
- runtime::RuntimeEstimate
- runtime::RuntimeFlopsEstimate
- runtime::RuntimeProfile
- runtime::ThroughputEstimate
- scratch::Scratch
- sphere5d::NeuronPoint
- sphere5d::Sphere5D
- tensor::TensorView
- trainer::DenseSgdConfig
Enums
- activations::ActivationKind
- attention::AttentionError
- attention::AttentionMask
- batching::BatchError
- beam_search::BeamError
- embeddings::EmbeddingError
- engine::ForwardError
- gradients::GradientError
- inference::InferenceError
- initializers::InitError
- initializers::InitKind
- kv_cache::KvCacheError
- layers::LayerError
- layers::LayerSpec
- lora::LoraError
- losses::LossError
- losses::LossKind
- metrics::MetricError
- model_config::ConfigError
- model_format::ModelFormatError
- moe::MoeError
- normalization::NormError
- optimizers::OptimizerError
- optimizers::OptimizerKind
- quantization::QuantError
- rnn_api::RnnApiError
- rnn_format::Error
- rope::RopeError
- runtime::RuntimeError
- sampling::SamplingError
- schedulers::LrSchedule
- schedulers::ScheduleError
- sphere5d::SphereError
- trainer::TrainError
Functions
- activations::activation_lut
- activations::apply_activation_slice
- activations::derivative_is_finite
- activations::validate_activation_outputs
- attention::scaled_dot_product_attention
- batching::count_non_pad
- batching::for_each_token_row
- batching::make_padding_mask
- batching::max_sequence_len
- batching::pad_sequences_u32
- batching::sequence_lengths
- beam_search::log_softmax_in_place
- beam_search::prune_by_threshold
- beam_search::select_top_beams
- conv3d::conv3d_forward
- conv3d::conv3d_is_compatible
- conv3d::conv3d_layout_compatible
- conv3d::conv3d_output_dim
- conv3d::conv3d_output_shape
- conv5d::conv5d_backward
- conv5d::conv5d_forward
- conv5d::conv5d_is_compatible
- conv5d::conv5d_layout_compatible
- conv5d::conv5d_output_dim
- conv5d::conv5d_output_shape
- crypto::constant_time_eq
- crypto::digest_to_hex_lower
- crypto::sha256_bytes
- crypto::sha512_bytes
- crypto::verify_sha256
- crypto::verify_sha512
- embeddings::gather_embeddings
- embeddings::tied_output_projection
- engine::forward_dense_plan
- engine::forward_dense_plan_big_kernel
- engine::required_batch_scratch_len
- engine::required_single_infer_scratch
- engine::validate_forward_io
- gradients::all_finite
- gradients::clip_by_global_norm
- gradients::has_inf
- gradients::has_nan
- gradients::l2_norm
- gradients::within_abs_bound
- inference::argmax_index
- inference::forward_dense_batch
- inference::normalize_logits_in_place
- inference::softmax_stable
- initializers::expected_parameter_counts
- initializers::init_kind_is_finite
- initializers::initialize_dense_parameters
- initializers::layers_are_valid
- initializers::mean
- initializers::min_max
- initializers::params_all_finite
- initializers::variance
- layers::build_dense_specs_from_layers
- layers::dense_chain_widths
- layers::dense_desc
- layers::layer_chain_is_compatible
- layers::total_bias_count
- layers::total_weight_count
- lora::apply_lora_delta
- losses::loss_and_gradient
- losses::reduce_mean
- losses::reduce_sum
- math::cosf
- math::expf
- math::lnf
- math::powf
- math::roundf
- math::sinf
- math::sqrtf
- math::tanhf
- metrics::accuracy_top1_from_one_hot
- metrics::argmax
- metrics::cross_entropy_from_probabilities
- metrics::mae
- metrics::mse
- model_config::base_transformer
- model_config::small_transformer
- model_config::tiny_transformer
- model_format::decode_dense_model_v1
- model_format::encode_dense_model_v1
- model_format::encoded_size_v1
- model_format::expected_encoded_size_from_header
- model_format::has_full_payload
- model_format::is_v1_header_consistent
- model_format::parse_header_v1
- moe::route_top1
- moe::top1_gating
- network::network_stats
- network::validate_network_parts
- normalization::layer_norm
- normalization::layer_norm_in_place
- normalization::rms_norm
- normalization::rms_norm_in_place
- optimizers::apply_optimizer_step
- optimizers::optimizer_state_len
- profiler::arithmetic_intensity
- profiler::bytes_per_second
- profiler::has_recorded_work
- profiler::is_memory_heavy
- profiler::ops_per_second
- quantization::dequantize_i8_symmetric
- quantization::matmul_i8_f32
- quantization::quantize_i8_symmetric
- rnn_api::rnn_dense_required_buffers
- rnn_api::rnn_dense_required_infer_scratch_from_specs
- rnn_api::rnn_pack_dense_v1
- rnn_api::rnn_required_dense_from_bytes_v1
- rnn_api::rnn_required_dense_from_topology
- rnn_api::rnn_run_dense_v1
- rnn_api::rnn_unpack_dense_v1
- rnn_api::rnn_validate_dense_counts
- rnn_api::rnn_validate_dense_topology
- rnn_format::all_blob_bounds_valid
- rnn_format::blob_bounds_valid
- rnn_format::find_blob_index
- rnn_format::parse_rnn_from_bytes
- rnn_format::total_blob_payload_bytes
- rope::apply_rope_in_place
- runtime::check_runtime_budget
- runtime::estimate_runtime_flops
- runtime::estimate_runtime_memory
- runtime::estimate_tokens_per_second
- runtime::fit_from_estimate
- sampling::argmax_sample
- sampling::sample_from_cumulative
- sampling::softmax_temperature
- sampling::top_k_mask
- sampling::top_p_cutoff
- schedulers::compute_learning_rate
- scratch::alloc_f32_slice
- scratch::alloc_u32_slice
- scratch::usage_bytes
- scratch::usage_ratio
- sphere5d::centroid
- sphere5d::layer_bounds
- sphere5d::mean_radius
- sphere5d::neuron_exists
- tensor::tensor_add_in_place
- tensor::tensor_fill
- tensor::tensor_scale_in_place
- trainer::config_is_valid
- trainer::dense_sgd_epoch
- trainer::dense_sgd_step
- trainer::io_matches_layers
- trainer::required_train_buffer_len