All modules for which code is available

• collections
• torch._tensor
• torch.nn.modules.sparse
• torch.optim.adam
• torch.optim.adamw
• torch.optim.sgd
• typing
• unit_scaling._modules
• unit_scaling.analysis
• unit_scaling.constraints
• unit_scaling.core.functional
• unit_scaling.formats
• unit_scaling.functional
• unit_scaling.optim
• unit_scaling.parameter
• unit_scaling.scale
• unit_scaling.transforms._compile
• unit_scaling.transforms._simulate_format
• unit_scaling.transforms._track_scales
• unit_scaling.transforms._unit_scale
• unit_scaling.transforms.utils
• unit_scaling.utils