ademamix
optimizer & lr scheduler & loss function collections in PyTorch
Modules, operations and models for computer vision in PyTorch