List of all items

Structs

• activations::GELU
• activations::LeakyReLU
• activations::Mish
• activations::ReLU
• activations::Sigmoid
• activations::Softmax
• activations::Swish
• activations::Tanh
• activations_minimal::GELU
• activations_minimal::ReLU
• activations_minimal::Sigmoid
• activations_minimal::Softmax
• activations_minimal::Tanh
• attention::alibi::AlibiConfig
• attention::alibi::AlibiSlopes
• attention::flash_v2::FlashAttentionV2
• attention::rope::RopeConfig
• attention::rope::RopeEmbedding
• attention::sliding_window::SlidingWindowAttention
• attention::sliding_window::SlidingWindowConfig
• attention::sparse::attention::SparseAttention
• attention::sparse::mask::AttentionMaskBuilder
• attention::sparse::types::SparseAttentionConfig
• attention::sparse::types::SparseAttentionMask
• attention::types::AttentionConfig
• attention::types::AttentionOutput
• callbacks::CallbackContext
• callbacks::CallbackManager
• callbacks::CosineAnnealingLR
• callbacks::EarlyStopping
• callbacks::FunctionCallback
• callbacks::GradientClipping
• callbacks::ModelCheckpoint
• callbacks::ReduceOnPlateau
• callbacks::ScirsMetricsCallback
• callbacks::StepDecay
• callbacks::TensorBoardLogger
• callbacks::VisualizationCallback
• data::CSVDataset
• data::ComposeAugmentation
• data::ComposeTransform
• data::DataLoader
• data::GaussianNoise
• data::GpuMemoryPool
• data::InMemoryDataset
• data::KFold
• data::MinMaxScaler
• data::OneHotEncoder
• data::PoolStatistics
• data::PooledBuffer
• data::RandomErasing
• data::RandomHorizontalFlip
• data::SizeClass
• data::StandardScaler
• data::SubsetDataset
• data::TransformedDataset
• distillation::DistillationConfig
• distillation::DistillationResult
• distillation::DistillationStatistics
• distillation::DistillationTrainer
• export::onnx::OnnxGraph
• export::onnx::OnnxModel
• export::onnx::OnnxNode
• export::onnx::OnnxTensor
• export::onnx::OnnxValueInfo
• export::weights::WeightStore
• inference::block_manager::BlockManager
• inference::block_manager::BlockManagerConfig
• inference::block_manager::SharedPrefixCache
• inference::kv_page::KvPage
• inference::kv_page::KvPageConfig
• inference::kv_page::KvPagePool
• inference::paged_attention::PagedAttentionConfig
• inference::paged_attention::PagedAttentionForward
• inference::speculative::rejection_sampling::SpeculativeDecoder
• inference::speculative::rejection_sampling::TokenDist
• inference::speculative::rejection_sampling::Xorshift64
• inference::speculative::types::SpeculativeConfig
• inference::speculative::types::SpeculativeResult
• layers::ActivityRegularization
• layers::AttentionConfig
• layers::Embedding
• layers::EmbeddingConfig
• layers::FlashAttention
• layers::FlashAttentionConfig
• layers::FlashAttentionV2
• layers::FlashAttentionV2Config
• layers::GqaKvCache
• layers::GroupedQueryAttention
• layers::GroupedQueryAttentionConfig
• layers::KvCache
• layers::L1ActivityRegularization
• layers::L2ActivityRegularization
• layers::LayerInfo
• layers::MultiHeadAttention
• layers::MultiQueryAttention
• layers::MultiQueryAttentionConfig
• layers::PositionalEmbedding
• layers::RotaryPositionEmbedding
• layers::SelfAttention
• layers::Sequential
• layers::conv::conv2d::Conv2D
• layers::conv::pooling::AvgPool2D
• layers::conv::pooling::GlobalAvgPool2D
• layers::conv::pooling::MaxPool2D
• layers::dense::Dense
• layers::dropout::Dropout
• layers::graph_conv::GraphAttentionLayer
• layers::graph_conv::GraphConvLayer
• layers::graph_conv::GraphSageLayer
• layers::normalization::BatchNorm
• layers::normalization::LayerNorm
• layers::recurrent::bidirectional::Bidirectional
• layers::recurrent::gru::GRU
• layers::recurrent::gru::GRUConfig
• layers::recurrent::lstm::LSTM
• layers::recurrent::lstm::LSTMConfig
• layers::recurrent::rnn::RNN
• layers::recurrent::rnn::RNNConfig
• layers::rnn_thread_safe::ThreadSafeBidirectional
• layers::rnn_thread_safe::ThreadSafeRNN
• lora::adalora::AdaLoraConfig
• lora::adalora::AdaLoraLayer
• lora::adapter::BottleneckAdapter
• lora::dora::DoraConfig
• lora::dora::DoraLinear
• lora::ia3::Ia3Adapter
• lora::ia3::Ia3Config
• lora::linear::LoRALinear
• lora::merge::QuantizedWeight
• lora::types::AdapterConfig
• lora::types::LoRAConfig
• lora::types::LoRAStats
• lora::vera::SharedRandomMatrices
• lora::vera::VeRAConfig
• lora::vera::VeRALayer
• lora::vera::VeraConfig
• lora::vera::VeraLayer
• losses::ContrastiveLoss
• losses::CrossEntropyLoss
• losses::FocalLoss
• losses::HuberLoss
• losses::MeanAbsoluteError
• losses::MeanSquaredError
• losses::TripletLoss
• models::architectures::bert::BertConfig
• models::architectures::bert::BertModel
• models::architectures::clip::CLIPConfig
• models::architectures::clip::CLIPTextConfig
• models::architectures::clip::CLIPTextEncoder
• models::architectures::convnext::ConvNeXtConfig
• models::architectures::convnext::ConvNeXtStageConfig
• models::architectures::efficientnet::EfficientNet
• models::architectures::efficientnet::EfficientNetConfig
• models::architectures::efficientnet::EfficientNetStage
• models::architectures::efficientnet::MBConvConfig
• models::architectures::fusion::BilinearFusion
• models::architectures::fusion::CrossModalAttention
• models::architectures::fusion::FeatureAlignment
• models::architectures::fusion::FeatureFusion
• models::architectures::fusion::FeatureFusionConfig
• models::architectures::fusion::FiLMModule
• models::architectures::gpt::GPTConfig
• models::architectures::gpt::GPTModel
• models::architectures::mamba::Mamba
• models::architectures::mamba::MambaBlock
• models::architectures::mamba::MambaConfig
• models::architectures::mamba::S4Layer
• models::architectures::mamba::SelectiveSSM
• models::architectures::mlp_mixer::MLPMixer
• models::architectures::mlp_mixer::MLPMixerConfig
• models::architectures::mlp_mixer::MixerBlock
• models::architectures::mlp_mixer::MixerMLP
• models::architectures::mobilenet::ConvBlockConfig
• models::architectures::mobilenet::MobileNet
• models::architectures::mobilenet::MobileNetConfig
• models::architectures::resnet::ResNet
• models::architectures::resnet::ResNetConfig
• models::architectures::resnet::ResNetLayer
• models::architectures::seq2seq::Attention
• models::architectures::seq2seq::Seq2Seq
• models::architectures::seq2seq::Seq2SeqConfig
• models::architectures::seq2seq::Seq2SeqDecoder
• models::architectures::seq2seq::Seq2SeqEncoder
• models::architectures::vgg::VGG
• models::architectures::vgg::VGGConfig
• models::architectures::vit::ViTConfig
• models::sequential::Sequential
• models::trainer::History
• models::trainer::Trainer
• models::trainer::TrainingConfig
• nas::NASConfig
• nas::NeuralArchitectureSearch
• nas::SearchResult
• nas::architecture_encoding::GraphEncoding
• nas::architecture_encoding::NodeAttributes
• nas::architecture_encoding::NodeType
• nas::architecture_encoding::SequentialEncoding
• nas::controller::ControllerConfig
• nas::controller::NASController
• nas::enas::ENASController
• nas::enas::ENASTrainer
• nas::enas::SuperNetwork
• nas::evaluator::ArchitectureEvaluator
• nas::evaluator::HardwareAwareEvaluator
• nas::evaluator::HardwareConfig
• nas::evaluator::MetricsConfig
• nas::gdas::GdasConfig
• nas::gdas::GdasSearch
• nas::hardware_aware::EnergyPredictor
• nas::hardware_aware::HardwareAwareSearch
• nas::hardware_aware::HardwareConstraints
• nas::hardware_aware::LatencyPredictor
• nas::hardware_aware::MemoryPredictor
• nas::multi_objective::MultiObjectiveConfig
• nas::multi_objective::MultiObjectiveOptimizer
• nas::multi_objective::MultiObjectiveSolution
• nas::multi_objective::Objective
• nas::performance_estimation::EarlyStoppingEstimator
• nas::performance_estimation::LearningCurveEstimator
• nas::performance_estimation::MultiFidelityEstimator
• nas::performance_estimation::PredictorNetworkEstimator
• nas::performance_estimation::SuperNetEstimator
• nas::performance_estimation::ZeroCostEstimator
• nas::predictor::ArchFeatureExtractor
• nas::predictor::GpPredictor
• nas::predictor::MlpPredictor
• nas::predictor::PredictorNasConfig
• nas::predictor::PredictorNasSearch
• nas::progressive_search::ProgressiveConfig
• nas::progressive_search::ProgressiveSearch
• nas::progressive_search::ProgressiveSearchBuilder
• nas::search_algorithms::BayesianOptimization
• nas::search_algorithms::DifferentiableSearch
• nas::search_algorithms::EvolutionarySearch
• nas::search_algorithms::RandomSearch
• nas::search_algorithms::ReinforcementSearch
• nas::search_space::Architecture
• nas::search_space::ConnectionMatrix
• nas::search_space::LayerChoice
• nas::search_space::SearchSpace
• nas::search_space::SearchSpaceConfig
• nas::snas::SnasConfig
• nas::snas::SnasSearch
• optimizers::adagrad::Adagrad
• optimizers::adam::Adam
• optimizers::adamw::AdamW
• optimizers::lr_scheduler_wrapper::LRSchedulerOptimizer
• optimizers::momentum::MomentumOptimizer
• optimizers::radam::RAdam
• optimizers::rmsprop::RMSprop
• optimizers::sgd::SGD
• quantization::DynamicQuantizer
• quantization::MixedBitWidthQuantizer
• quantization::PostTrainingQuantizer
• quantization::QuantizationAwareTraining
• quantization::QuantizationConfig
• quantization::QuantizationParams
• quantization::QuantizedTensor
• quantization::calibration::CalibrationRunner
• quantization::calibration::MinMaxObserver
• quantization::calibration::MovingAverageObserver
• quantization::calibration::ObserverStats
• quantization::calibration::PercentileObserver
• quantization::gptq::GptqConfig
• quantization::gptq::GptqQuantizer
• quantization::gptq::GptqResult
• quantization::int4::Int4GroupParams
• quantization::int4::Int4Tensor
• quantization::int8::Int8Params
• quantization::int8::Int8Tensor
• serialization::ActivationFactory
• serialization::BatchNormConfig
• serialization::Conv2DConfig
• serialization::DenseConfig
• serialization::DropoutConfig
• serialization::LayerNormConfig
• serialization::MaxPool2DConfig
• serialization::SerializedModel
• serialization::architecture::ArchitectureConfig
• serialization::architecture::SerializableBertConfig
• serialization::architecture::SerializableEfficientNetConfig
• serialization::architecture::SerializableGPTConfig
• serialization::architecture::SerializableLayerInfo
• serialization::architecture::SerializableMambaConfig
• serialization::architecture::SerializableMobileNetConfig
• serialization::architecture::SerializableResNetConfig
• serialization::architecture::SerializableResNetLayer
• serialization::architecture::SerializableSequentialConfig
• serialization::safetensors::SafeTensorsHeader
• serialization::safetensors::SafeTensorsHeaderEntry
• serialization::safetensors::SafeTensorsReader
• serialization::safetensors::SafeTensorsWriter
• serialization::traits::ModelMetadata
• serialization::traits::NamedParameters
• serialization::traits::TensorInfo
• speculative::decoder::SpeculativeDecoder
• speculative::draft::NGramDraftModel
• speculative::draft::UniformDraftModel
• speculative::draft::Xorshift64
• speculative::types::DecodingStats
• speculative::types::SpeculativeConfig
• speculative::types::TokenDistribution
• speculative::types::VerificationResult
• speculative::verifier::SpeculativeVerifier
• tracing::executor::GraphExecutor
• tracing::graph_builder::GraphBuilder
• tracing::graph_builder::Tensor
• tracing::types::OpNode
• tracing::types::StaticGraph
• tracing::types::TensorSpec
• tracing::types::TraceConfig
• training::TrainingConfig
• training::TrainingSession
• training::ValidationSettings
• training::augmentation::AugmentationPipeline
• training::backprop_efficient::ActivationCheckpoint
• training::backprop_efficient::CheckpointStatistics
• training::backprop_efficient::CpuActivationStore
• training::backprop_efficient::EfficientBackprop
• training::backprop_efficient::GradientCheckpointManager
• training::checkpoint::CheckpointConfig
• training::checkpoint::CheckpointManager
• training::checkpoint::CheckpointMetadata
• training::checkpoint::LrSchedulerState
• training::checkpoint::OptimizerCheckpointState
• training::checkpoint::OptimizerStateMetadata
• training::checkpoint::ParamGroupState
• training::checkpoint::TrainingCheckpoint
• training::curriculum::CurriculumConfig
• training::curriculum::CurriculumConfigBuilder
• training::curriculum::CurriculumLearner
• training::curriculum::CurriculumSchedule
• training::curriculum::LossBasedScorer
• training::curriculum::StaticScorer
• training::early_stopping::EarlyStopping
• training::early_stopping::EarlyStoppingWithState
• training::early_stopping::StepResult
• training::enhanced_trainer::EarlyStoppingConfig
• training::enhanced_trainer::EnhancedTrainer
• training::enhanced_trainer::EnhancedTrainingConfig
• training::enhanced_trainer::GradientAccumulationSettings
• training::enhanced_trainer::LRWarmupConfig
• training::enhanced_trainer::OperationTiming
• training::enhanced_trainer::OptimizationAnalyzer
• training::enhanced_trainer::OptimizationRecommendation
• training::enhanced_trainer::ProfilingConfig
• training::enhanced_trainer::ProfilingResults
• training::enhanced_trainer::ProgressConfig
• training::enhanced_trainer::TrainingState
• training::enhanced_trainer::ValidationConfig
• training::federated::ClientUpdate
• training::federated::DifferentialPrivacyConfig
• training::federated::FederatedConfig
• training::federated::FederatedConfigBuilder
• training::federated::FederatedServer
• training::federated::GradientCompressionConfig
• training::federated::RoundStats
• training::gradient_accumulation::GradientAccumulationConfig
• training::gradient_accumulation::GradientAccumulator
• training::gradient_accumulation::GradientStats
• training::gradient_checkpointing::GradientCheckpointingConfig
• training::gradient_clipping::AdaptiveGradClipConfig
• training::gradient_clipping::GradientClipResult
• training::hparam_tuner::HParamSpace
• training::hparam_tuner::HParamTuner
• training::hparam_tuner::TrialResult
• training::lr_finder::LRFinder
• training::lr_finder::LRFinderConfig
• training::lr_finder::LRFinderConfigBuilder
• training::lr_finder::LRFinderPoint
• training::lr_finder::LRFinderResult
• training::lr_finder::TypedLRFinder
• training::metrics_tracker::BestMetric
• training::metrics_tracker::MetricEntry
• training::metrics_tracker::MetricHistory
• training::metrics_tracker::MetricStats
• training::metrics_tracker::MetricsTracker
• training::metrics_tracker::TrainingHistory
• training::mixed_precision::AutoMixedPrecision
• training::mixed_precision::GradScaler
• training::mixed_precision::GradScalerStats
• training::mixed_precision::Half
• training::mixed_precision::MasterWeights
• training::mixed_precision::MixedPrecisionCallback
• training::mixed_precision::MixedPrecisionConfig
• training::mixed_precision::MixedPrecisionConfigBuilder
• training::mixed_precision::MixedPrecisionStepResult
• training::mixed_precision::MixedPrecisionTensor
• training::mixed_precision::MixedPrecisionTrainer
• training::mixed_precision::TrainingStats
• training::optimized_dataloader::BatchSizeOptimizationResult
• training::optimized_dataloader::BatchSizeOptimizer
• training::optimized_dataloader::LoadingStats
• training::optimized_dataloader::MemoryAwareConfig
• training::optimized_dataloader::MemoryAwareDataLoader
• training::optimized_dataloader::MemoryAwarePrefetchIter
• training::optimized_dataloader::OptimizedDataLoader
• training::optimized_dataloader::OptimizedLoaderConfig
• training::optimized_dataloader::PrefetchingIterator
• training::pipeline_parallel::MicroBatch
• training::pipeline_parallel::PipelineConfig
• training::pipeline_parallel::PipelineParallel
• training::pipeline_parallel::PipelineStage
• training::profiler::BatchStats
• training::profiler::Bottleneck
• training::profiler::EpochStats
• training::profiler::LayerProfile
• training::profiler::ProfileSummary
• training::profiler::ScopedTimer
• training::profiler::TrainingProfiler
• training::progress_monitor::ProgressMonitorConfig
• training::quantization_aware::QuantizationAwareConfig
• training::schedulers::ChainedScheduler
• training::schedulers::CosineAnnealingLR
• training::schedulers::CosineAnnealingWarmRestarts
• training::schedulers::CyclicLR
• training::schedulers::ExponentialLR
• training::schedulers::LinearWarmup
• training::schedulers::MultiStepLR
• training::schedulers::OneCycleLR
• training::schedulers::PolynomialLR
• training::schedulers::ReduceOnPlateau
• training::schedulers::StepLR
• training::schedulers::WarmupCosine
• training::sparse_training::SparseTrainingConfig
• training::tensor_parallel::ColumnParallelLinear
• training::tensor_parallel::ParallelEmbedding
• training::tensor_parallel::RowParallelLinear
• training::tensor_parallel::TensorParallelConfig
• transformer::FeedForward
• transformer::Transformer
• transformer::TransformerConfig
• transformer::TransformerDecoder
• transformer::TransformerDecoderLayer
• transformer::TransformerEncoder
• transformer::TransformerEncoderLayer
• utils::colors::ColorOptions
• utils::datasets::BatchIterator
• utils::datasets::DataLoader
• utils::datasets::Dataset
• utils::evaluation::ConfusionMatrix
• utils::evaluation::FeatureImportance
• utils::evaluation::LearningCurve
• utils::evaluation::ROCCurve
• utils::metrics::BinaryAccuracy
• utils::metrics::CategoricalAccuracy
• utils::metrics::MeanSquaredError
• utils::metrics::R2Score
• utils::model_viz::ModelVizOptions
• utils::positional_encoding::LearnedPositionalEncoding
• utils::positional_encoding::PositionalEncodingFactory
• utils::positional_encoding::RelativePositionalEncoding
• utils::positional_encoding::RotaryPositionalEncoding
• utils::positional_encoding::SinusoidalPositionalEncoding
• visualization::VisualizationConfigBuilder
• visualization::VisualizationSuite
• visualization::activations::ActivationHistogram
• visualization::activations::ActivationStatistics
• visualization::activations::ActivationVisualizationOptions
• visualization::activations::ActivationVisualizer
• visualization::activations::FeatureMapInfo
• visualization::attention::AttentionData
• visualization::attention::AttentionStatistics
• visualization::attention::AttentionVisualizationOptions
• visualization::attention::AttentionVisualizer
• visualization::attention::CompressionSettings
• visualization::attention::ExportOptions
• visualization::attention::HeadInfo
• visualization::attention::HighlightConfig
• visualization::attention::LayerInfo
• visualization::attention::Resolution
• visualization::config::CustomTheme
• visualization::config::FontConfig
• visualization::config::GridConfig
• visualization::config::InteractiveConfig
• visualization::config::LayoutConfig
• visualization::config::Margins
• visualization::config::PerformanceConfig
• visualization::config::StyleConfig
• visualization::config::VisualizationConfig
• visualization::network::BoundingBox
• visualization::network::Connection
• visualization::network::ConnectionVisualProps
• visualization::network::DataFlowInfo
• visualization::network::LayerIOInfo
• visualization::network::LayerInfo
• visualization::network::LayerPosition
• visualization::network::LayerVisualProps
• visualization::network::NetworkLayout
• visualization::network::NetworkVisualizer
• visualization::network::Point2D
• visualization::network::Size2D
• visualization::network::ThroughputInfo
• visualization::training::AxisConfig
• visualization::training::LineStyleConfig
• visualization::training::MarkerConfig
• visualization::training::PlotConfig
• visualization::training::SeriesConfig
• visualization::training::SystemMetrics
• visualization::training::TickConfig
• visualization::training::TrainingMetrics
• visualization::training::TrainingVisualizer

Enums

• attention::sparse::types::SparsePattern
• attention::types::AttentionMask
• attention::types::PositionEncoding
• callbacks::CallbackTiming
• callbacks::GradientClippingMethod
• callbacks::ScheduleMethod
• distillation::AttentionType
• distillation::DistanceMetric
• distillation::DistillationMethod
• distillation::EnsembleAggregation
• distillation::FeatureAdaptation
• distillation::RelationType
• error::NeuralError
• export::onnx::OnnxAttribute
• export::onnx::OnnxDataType
• export::weights::WeightFormat
• inference::InferenceError
• layers::AttentionMask
• layers::LayerConfig
• layers::conv::common::PaddingMode
• layers::graph_conv::GraphActivation
• layers::graph_conv::SageAggregator
• layers::recurrent::rnn::RecurrentActivation
• layers::rnn_thread_safe::RecurrentActivation
• lora::types::AdapterActivation
• lora::types::LoRATarget
• models::architectures::convnext::ConvNeXtVariant
• models::architectures::fusion::FusionMethod
• models::architectures::mobilenet::MobileNetVersion
• models::architectures::resnet::ResNetBlock
• models::architectures::seq2seq::AttentionType
• models::architectures::seq2seq::RNNCellType
• models::architectures::vgg::VGGVariant
• nas::gdas::TemperatureSchedule
• nas::hardware_aware::HardwarePlatform
• nas::hardware_aware::QuantizationSupport
• nas::multi_objective::MultiObjectiveAlgorithm
• nas::predictor::AcquisitionFunction
• nas::predictor::PredictorType
• nas::progressive_search::ExpansionStrategy
• nas::search_space::Choice
• nas::search_space::LayerType
• quantization::QuantizationMode
• quantization::QuantizationScheme
• quantization::int4::GroupSize
• quantization::int8::Granularity
• quantization::int8::Int8Mode
• serialization::ActivationFunction
• serialization::LayerConfig
• serialization::LayerType
• serialization::SerializationFormat
• serialization::safetensors::SafeTensorsDtype
• serialization::traits::ModelFormat
• tracing::types::DType
• tracing::types::OpAttr
• tracing::types::OpType
• training::augmentation::AugmentationType
• training::backprop_efficient::RecomputationPolicy
• training::checkpoint::CheckpointError
• training::curriculum::CompetenceSchedule
• training::curriculum::CurriculumStrategy
• training::early_stopping::StopReason
• training::early_stopping::StoppingMode
• training::enhanced_trainer::RecommendationType
• training::enhanced_trainer::WarmupSchedule
• training::federated::AggregationMethod
• training::federated::ClientSelectionStrategy
• training::gradient_clipping::ClipNormType
• training::hparam_tuner::HParamValue
• training::hparam_tuner::SearchStrategy
• training::hparam_tuner::SpaceType
• training::lr_finder::LRFinderStatus
• training::lr_finder::LRScheduleType
• training::metrics_tracker::MetricGoal
• training::mixed_precision::AmpContextState
• training::mixed_precision::FP32Operation
• training::mixed_precision::LossScalingStrategy
• training::mixed_precision::PrecisionMode
• training::pipeline_parallel::PipelineSchedule
• training::profiler::ProfilePhase
• training::schedulers::CyclicMode
• utils::colors::Color
• utils::colors::Style
• utils::datasets::Normalization
• utils::initializers::Initializer
• utils::positional_encoding::PositionalEncodingType
• visualization::activations::ActivationNormalization
• visualization::activations::ActivationVisualizationType
• visualization::activations::ChannelAggregation
• visualization::activations::Colormap
• visualization::attention::AttentionVisualizationType
• visualization::attention::DataFormat
• visualization::attention::ExportFormat
• visualization::attention::ExportQuality
• visualization::attention::HeadAggregation
• visualization::attention::HeadSelection
• visualization::attention::HighlightStyle
• visualization::attention::VideoFormat
• visualization::config::ColorPalette
• visualization::config::DownsamplingStrategy
• visualization::config::ImageFormat
• visualization::config::Theme
• visualization::network::ArrowStyle
• visualization::network::ConnectionType
• visualization::network::LayoutAlgorithm
• visualization::network::LineStyle
• visualization::training::AxisScale
• visualization::training::LineStyle
• visualization::training::MarkerShape
• visualization::training::PlotType
• visualization::training::TickFormat
• visualization::training::UpdateMode

Traits

• activations::Activation
• activations_minimal::Activation
• callbacks::Callback
• callbacks::LearningRateScheduler
• data::Augmentation
• data::Dataset
• data::Transform
• error::StdError
• export::onnx::OnnxExportable
• layers::Layer
• layers::ParamLayer
• losses::Loss
• models::Model
• nas::architecture_encoding::ArchitectureEncoding
• nas::performance_estimation::PerformanceEstimator
• nas::search_algorithms::SearchAlgorithm
• optimizers::Optimizer
• optimizers::OptimizerStep
• quantization::calibration::Observer
• serialization::model_serializer::ModelSerializer
• serialization::traits::ExtractParameters
• serialization::traits::ModelDeserialize
• serialization::traits::ModelSerialize
• speculative::draft::DraftModel
• speculative::verifier::TargetModel
• training::curriculum::DifficultyScorer
• training::schedulers::LRScheduler
• utils::metrics::Metric
• utils::positional_encoding::PositionalEncoding

Functions

• attention::flash_v2::naive_attention
• autograd::autograd_example
• data::create_batches
• data::iter_batches
• data::train_val_split
• distillation::distill
• export::onnx::export_activation
• export::onnx::export_batchnorm
• export::onnx::export_conv2d
• export::onnx::export_linear
• export::onnx::export_sequential
• export::weights::pytorch_to_scirs2_names
• export::weights::scirs2_to_pytorch_names
• layers::conv::common::calculate_adaptive_pooling_params
• layers::conv::common::calculate_outputshape
• layers::conv::common::validate_conv_params
• layers::flash_attention_compute
• layers::flash_attention_v2_compute
• linalg::adam_update
• linalg::adaptive_avg_pool2d
• linalg::adaptive_avg_pool2d_backward
• linalg::batch_matmul
• linalg::batch_matmul_backward
• linalg::batch_norm_backward
• linalg::batch_norm_forward
• linalg::batch_vecmat
• linalg::clip_grad_norm
• linalg::col2im
• linalg::conv2d
• linalg::conv2d_backward
• linalg::dropout
• linalg::dropout_backward
• linalg::im2col
• linalg::layer_norm
• linalg::layer_norm_backward
• linalg::log_softmax
• linalg::lstm_cell
• linalg::max_pool2d
• linalg::max_pool2d_backward
• lora::merge::compute_effective_weight
• lora::merge::merge_lora_weights
• lora::merge::merge_multiple_lora
• lora::merge::quantize_merged_weight
• models::architectures::efficientnet::swish
• models::architectures::mobilenet::hard_sigmoid
• models::architectures::mobilenet::hard_swish
• models::architectures::mobilenet::relu6
• optimizers::lr_scheduler_wrapper::with_cosine_annealing
• optimizers::lr_scheduler_wrapper::with_step_decay
• quantization::calibration::minmax_calibration_runner
• quantization::gptq::naive_quantize
• quantization::int4::dequantize_int4
• quantization::int4::int4_compression_ratio
• quantization::int4::int4_memory_bytes
• quantization::int4::int4_roundtrip_mse
• quantization::int4::quantize_int4
• quantization::int8::dequantize_int8
• quantization::int8::quantize_int8
• quantization::int8::quantized_matmul
• quantization::int8::roundtrip_mse
• quantization::quantize_model
• quantization::utils::compute_quantization_error
• quantization::utils::convert_quantization_scheme
• quantization::utils::estimate_performance_gain
• quantization::utils::estimate_size_reduction
• serialization::architecture::detect_architecture
• serialization::architecture::detect_architecture_from_bytes
• serialization::load_model
• serialization::model_serializer::load_bert
• serialization::model_serializer::load_resnet
• serialization::model_serializer::named_parameters_to_map
• serialization::model_serializer::save_bert
• serialization::model_serializer::save_resnet
• serialization::safetensors::read_named_parameters
• serialization::safetensors::validate_safetensors_file
• serialization::safetensors::write_named_parameters
• serialization::save_model
• speculative::draft::build_ngram_table
• speculative::verifier::compute_adjusted_distribution
• tensor_ops::concat
• tensor_ops::dropout
• tensor_ops::global_avg_pool
• tensor_ops::max_pool2d
• tensor_ops::upsample_bilinear
• tensor_ops::upsample_nearest
• tracing::executor::optimize
• training::augmentation::apply_cutmix
• training::augmentation::apply_mixup
• training::checkpoint::best_checkpoint
• training::checkpoint::checkpoint_dir_name
• training::checkpoint::latest_checkpoint
• training::checkpoint::list_checkpoints
• training::checkpoint::load_checkpoint
• training::checkpoint::save_checkpoint
• training::federated::clip_l2_norm
• training::gradient_clipping::clip_grad_adaptive
• training::gradient_clipping::clip_grad_agc
• training::gradient_clipping::clip_grad_norm
• training::gradient_clipping::clip_grad_value
• training::gradient_clipping::grad_norm
• training::lr_finder::find_optimal_lr
• training::mixed_precision::clip_by_global_norm
• training::mixed_precision::clip_by_value
• training::mixed_precision::contains_inf_or_nan
• training::mixed_precision::global_norm
• training::mixed_precision::tensor_norm
• training::profiler::estimate_conv2d_memory
• training::profiler::estimate_dense_flops
• training::profiler::estimate_dense_memory
• utils::colors::color_legend
• utils::colors::colored_metric_cell
• utils::colors::colored_string
• utils::colors::colorize
• utils::colors::colorize_and_style
• utils::colors::colorize_bg
• utils::colors::gradient_color
• utils::colors::heatmap_cell
• utils::colors::heatmap_color_legend
• utils::colors::heatmap_gradient_color
• utils::colors::stylize
• utils::colors::supports_color
• utils::datasets::normalize_features
• utils::datasets::one_hot_encode
• utils::initializers::compute_fan
• utils::initializers::init_weights
• utils::initializers::kaiming_normal
• utils::initializers::kaiming_uniform
• utils::initializers::orthogonal
• utils::initializers::xavier_normal
• utils::initializers::xavier_uniform
• utils::metrics::binary_accuracy
• utils::metrics::categorical_accuracy
• utils::metrics::mean_squared_error
• utils::model_viz::sequential_model_dataflow
• utils::model_viz::sequential_model_summary
• utils::one_hot_encode
• utils::random_normal
• utils::train_test_split
• visualization::utils::dashboard_config
• visualization::utils::publication_config
• visualization::utils::quick_config

Type Aliases

• error::Error
• error::Result
• inference::InferenceResult
• inference::block_manager::SeqId
• inference::kv_page::PageId
• layers::conv::common::MaxIndicesCache
• layers::conv::common::MaxIndicesCache3D
• layers::recurrent::GruForwardOutput
• layers::recurrent::GruGateCache
• layers::recurrent::GruStateCache
• layers::recurrent::LstmGateCache
• layers::recurrent::LstmStateCache
• layers::recurrent::LstmStepOutput
• layers::recurrent::RnnStateCache
• losses::SmoothL1Loss
• nas::evaluator::EvaluationMetrics
• training::optimized_dataloader::BatchResult
• utils::TrainTestSplitResult
• visualization::ActivationViz
• visualization::AttentionViz
• visualization::NetworkViz
• visualization::TrainingViz

Constants

• utils::colors::RESET