torch/optim/adamw.pyi

from typing import Tuple
from .optimizer import _params_t, Optimizer

class AdamW(Optimizer):
    def __init__(self, params: _params_t, lr: float=..., betas: Tuple[float, float]=..., eps: float=..., weight_decay: float=..., amsgrad: bool = ...) -> None: ...