API Reference¶

Full documentation for every public class and function in imbalanced_losses.

Losses¶

Class	Module	Description
`SigmoidFocalLoss`	`imbalanced_losses`	Binary / multi-label focal loss
`SoftmaxFocalLoss`	`imbalanced_losses`	Multiclass focal loss with softmax
`SmoothAPLoss`	`imbalanced_losses`	Differentiable Average Precision loss
`RecallAtQuantileLoss`	`imbalanced_losses`	Differentiable Recall-at-Quantile loss
`PAUCAtBudgetLoss`	`imbalanced_losses`	Differentiable partial-AUC over an FPR band

Class / Function	Module	Description
`LossWarmupWrapper`	`imbalanced_losses`	Phase-switching warmup + temperature decay
`all_gather_with_grad`	`imbalanced_losses.distributed`	All-gather preserving gradient flow
`all_gather_no_grad`	`imbalanced_losses.distributed`	All-gather without gradient tracking