optax.schedules.Schedule# optax.schedules.Schedule# alias of Callable[[Array | ndarray | bool | number | bool | int | float | complex], Array | ndarray | bool | number | bool | int | float | complex]
