StepLR

Auto-generated from jax_code_defs

PyTorch

API: torch.optim.lr_scheduler.StepLR
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: optax.exponential_decay
Strategy: Plugin (scheduler_rewire)