# Taken from pytorch-optimizer : https://github.com/jettify/pytorch-optimizer
from typing import Iterable, Union, Callable, Dict, Optional, Tuple, Any
from torch import Tensor

Params = Union[Iterable[Tensor], Iterable[dict]]

LossClosure = Callable[[], float]
OptLossClosure = Optional[LossClosure]
Betas2 = Tuple[float, float]
State = Dict[str, Any]
OptFloat = Optional[float]
Nus2 = Tuple[float, float]