List of all items
Structs
- AccumulationConfig
- AccumulationStats
- Accuracy
- AdaBeliefOptimizer
- AdaMaxOptimizer
- AdagradOptimizer
- AdamOptimizer
- AdamPOptimizer
- AdamWOptimizer
- AttentionTransferLoss
- AugRng
- AugStats
- AugmentationPipeline
- AutocastContext
- AveragingEnsemble
- BCEWithLogitsLoss
- BaggingHelper
- BalancedAccuracy
- BatchCallback
- BatchConfig
- BatchIterator
- BatchReweighter
- BayesianOptimization
- CallbackList
- CheckpointCallback
- ClassBalancedSampler
- CohensKappa
- CompetenceCurriculum
- CompositeAugmenter
- CompositeRegularization
- ConfusionMatrix
- ConsoleLogger
- ConstraintViolationLoss
- ContrastiveLoss
- CosineAnnealingLrScheduler
- CosineAnnealingScheduler
- CrossEntropyLoss
- CrossValidationResults
- CsvLoader
- CsvLogger
- CurriculumManager
- CurriculumSampler
- CutMixAugmenter
- CutOutAugmenter
- CyclicLrScheduler
- CyclicalScheduler
- DataPreprocessor
- DataShuffler
- Dataset
- DiceCoefficient
- DiceLoss
- DiscriminativeFineTuning
- DistillationLoss
- DropBlock
- DropPath
- DynamicRangeCalibrator
- EarlyStoppingCallback
- ElasticNetRegularization
- EpisodeSampler
- EpochCallback
- ExpectedCalibrationError
- ExponentialCurriculum
- ExponentialLrScheduler
- ExponentialStochasticDepth
- F1Score
- FeatureDistillationLoss
- FeatureExtractorMode
- FewShotAccuracy
- FileLogger
- FocalLoss
- FocalSampler
- GaussianProcess
- GcConfig
- GcStats
- GlobalPruner
- GradientAccumulationCallback
- GradientAccumulationStats
- GradientAccumulator
- GradientBuffer
- GradientCentralization
- GradientCheckpointConfig
- GradientMonitor
- GradientPruner
- GradientScaler
- GradientStats
- GradientSummary
- GridSearch
- GroupLassoRegularization
- HardNegativeMiner
- HingeLoss
- HistogramCallback
- HistogramStats
- HuberLoss
- HyperparamResult
- ImportanceSampler
- IoU
- JsonlLogger
- KFold
- KLDivergenceLoss
- L1Regularization
- L2Regularization
- LabelEncoder
- LabelSmoothingLoss
- LambOptimizer
- LarsOptimizer
- LayerFreezingConfig
- LayerPruningStats
- LearningRateFinder
- LeaveOneOut
- LinearCurriculum
- LinearDropBlockScheduler
- LinearModel
- LinearStochasticDepth
- LionConfig
- LionOptimizer
- LogicalLoss
- LookaheadOptimizer
- LossConfig
- LrRangeTestAnalyzer
- MAML
- MAMLConfig
- MagnitudePruner
- MatchingNetwork
- MatthewsCorrelationCoefficient
- MaxNormRegularization
- MaximumCalibrationError
- MeanAveragePrecision
- MeanIoU
- MemoryBudgetManager
- MemoryEfficientTraining
- MemoryProfilerCallback
- MemorySettings
- MemoryStats
- MetaStats
- MetaTask
- MetricTracker
- MetricsLogger
- MixedPrecisionStats
- MixedPrecisionTrainer
- MixupAugmenter
- MixupLoss
- ModelEMACallback
- ModelSoup
- ModelSummary
- MseLoss
- MultiStepLrScheduler
- MultiTaskLoss
- NAdamOptimizer
- NoAugmentation
- NoamScheduler
- NoiseAugmenter
- NormalizedDiscountedCumulativeGain
- OneCycleLrScheduler
- OneCyclePolicyScheduler
- OneHotEncoder
- OnlineHardExampleMiner
- OptimizerConfig
- OrthogonalRegularization
- PCGrad
- ParameterDifference
- ParameterStats
- PerClassMetrics
- PolyLoss
- PolynomialDecayLrScheduler
- Precision
- ProdigyConfig
- ProdigyOptimizer
- ProfilingCallback
- ProfilingStats
- ProgressiveUnfreezing
- PrototypicalDistance
- PruningConfig
- PruningStats
- QuantizationAwareTraining
- QuantizationConfig
- QuantizationParams
- QuantizedTensor
- Quantizer
- RAdamOptimizer
- RMSpropOptimizer
- RandomErasingAugmenter
- RandomSearch
- Recall
- ReduceLROnPlateauScheduler
- ReduceLrOnPlateauCallback
- Reptile
- ReptileConfig
- RocCurve
- RotationAugmenter
- RuleSatisfactionLoss
- SWACallback
- SamOptimizer
- ScaleAugmenter
- ScheduleFreeAdamW
- ScheduleFreeConfig
- SchedulerConfig
- SelfPacedCurriculum
- SgdOptimizer
- SgdrScheduler
- SophiaConfig
- SophiaOptimizer
- SpectralNormalization
- StackingEnsemble
- StepDecayScheduler
- StepLrScheduler
- StratifiedKFold
- StructuredPruner
- SupportSet
- TaskCurriculum
- TensorBoardLogger
- TimeEstimator
- TimeSeriesSplit
- TopKAccuracy
- Trainer
- TrainerConfig
- TrainingCheckpoint
- TrainingHistory
- TrainingState
- TransferLearningManager
- TripletLoss
- TverskyLoss
- ValidationCallback
- VotingEnsemble
- WarmupCosineLrScheduler
- WarmupScheduler
- adversarial::AdversarialExample
- adversarial::AdversarialTrainStats
- adversarial::AttackConfig
- adversarial::CrossEntropyAttackLoss
- adversarial::LinearAttackModel
- adversarial::MseAttackLoss
- checkpoint::CheckpointManager
- checkpoint::CheckpointMetadata
- checkpoint::LossTracker
- checkpoint::OptimizerCheckpoint
- checkpoint::ParamState
- early_stopping::EarlyStoppingConfig
- early_stopping::EarlyStoppingMonitor
- early_stopping::MultiMetricMonitor
- early_stopping::PlateauDetector
- early_stopping::TrainingProgress
- lora::adapter::LayerStats
- lora::adapter::LoraAdapter
- lora::adapter::LoraAdapterSummary
- lora::config::LoraConfig
- lora::layer::LoraLayer
- neural_ode::AdaptiveSolution
- neural_ode::AdjointResult
- neural_ode::NeuralOde
- neural_ode::OdeSolution
- neural_ode::OdeSolverConfig
- online_learning::Ftrl
- online_learning::OnlineGradientDescent
- online_learning::OnlineStats
- online_learning::OnlineUpdateResult
- online_learning::PassiveAggressive
- online_learning::Perceptron
- weight_init::InitRng
- weight_init::InitStats
Enums
- AccumulationError
- AcquisitionFunction
- AugmentationError
- AugmentationStep
- BitWidth
- CheckpointCompression
- CheckpointStrategy
- CyclicLrMode
- DistanceMetric
- GcStrategy
- GpKernel
- GradClipMode
- GradientScalingStrategy
- Granularity
- HyperparamSpace
- HyperparamValue
- LossScaler
- MiningStrategy
- PlateauMode
- PrecisionMode
- PreprocessingMethod
- QuantizationMode
- ReweightingStrategy
- SchedulerError
- SchedulerType
- ShotType
- SophiaVariant
- SoupRecipe
- StructuredPruningAxis
- TaskWeightingStrategy
- TrainError
- VotingMode
- adversarial::AdversarialError
- adversarial::PerturbNorm
- checkpoint::CheckpointError
- checkpoint::CheckpointFormat
- early_stopping::EarlyStoppingDecision
- early_stopping::MonitorMode
- early_stopping::MultiMetricPolicy
- lora::error::LoraError
- neural_ode::OdeError
- online_learning::OGDLoss
- online_learning::OnlineError
- online_learning::PAVariant
- weight_init::FanMode
- weight_init::InitError
Traits
- AutodiffModel
- Callback
- CrossValidationSplit
- CurriculumStrategy
- DataAugmenter
- DynamicModel
- Ensemble
- LoggingBackend
- Loss
- LrScheduler
- LrSchedulerV2
- MetaLearner
- Metric
- Model
- Optimizer
- Pruner
- Regularizer
- adversarial::AttackLoss
- adversarial::AttackModel
- neural_ode::OdeFunc
- online_learning::OnlineLearner
Functions
- adversarial::adversarial_training_loss
- adversarial::fgsm
- adversarial::pgd
- adversarial::project_l1
- adversarial::project_l2
- adversarial::project_linf
- adversarial::robustness_eval
- center_crop_2d
- checkpoint::deserialize_checkpoint
- checkpoint::serialize_checkpoint
- clip
- compare_models
- compute_gradient_stats
- cutmix
- denormalize
- dropout
- dropout_mask
- extract_batch
- format_duration
- gaussian_noise
- mixup
- neural_ode::dopri5_solve
- neural_ode::rk4_solve
- normalize
- online_learning::online_evaluate
- print_gradient_report
- random_crop_2d
- random_hflip
- random_vflip
- weight_init::compute_fans
- weight_init::constant_init
- weight_init::gain_for_activation
- weight_init::kaiming_normal
- weight_init::kaiming_uniform
- weight_init::lecun_normal
- weight_init::lecun_uniform
- weight_init::normal_init
- weight_init::ones_init
- weight_init::orthogonal_init
- weight_init::uniform_init
- weight_init::xavier_normal
- weight_init::xavier_uniform
- weight_init::zeros_init