refactor: unify learning rate schedulers with array API #13596

Job	Run time
Build source distribution	14s
Build wheels for cp311-macosx_arm64	5m 30s
Build wheels for cp311-macosx_x86_64	12m 42s
Build wheels for cp311-win_amd64	5m 13s
Build wheels for cp311-manylinux_x86_64	7m 8s
Build wheels for cp310-manylinux_aarch64	6m 38s
build_pypi_index	11s
build_docker (12)	3m 26s
upload_pypi	0s
deploy_pypi_index	-1s
Pass testing build wheels	4s
	41m 5s

Provide feedback