refactor: unify learning rate schedulers with array API #13699

Job	Run time
Test Python (3, 3.10)	33m 55s
Test Python (1, 3.10)	38m 8s
Test Python (2, 3.13)	33m 17s
Test Python (1, 3.13)	47m 1s
Test Python (3, 3.13)	31m 31s
Test Python (2, 3.10)	33m 39s
Test Python (10, 3.13)	29m 38s
Test Python (4, 3.13)	30m 37s
Test Python (7, 3.13)	29m 59s
Test Python (8, 3.13)	43m 13s
Test Python (11, 3.10)	34m 8s
Test Python (5, 3.10)	32m 50s
Test Python (5, 3.13)	32m 55s
Test Python (12, 3.13)	31m 18s
Test Python (4, 3.10)	33m 29s
Test Python (7, 3.10)	34m 34s
Test Python (8, 3.10)	33m 30s
Test Python (10, 3.10)	35m 2s
Test Python (12, 3.10)	31m 47s
Test Python (9, 3.10)	32m 23s
Test Python (6, 3.13)	32m 4s
Test Python (6, 3.10)	44m 39s
Test Python (11, 3.13)	29m 32s
Test Python (9, 3.13)	30m 26s
Combine and update integration test durations (3.10)	7s
Combine and update integration test durations (3.13)	9s
Pass testing Python	4s
	13h 39m 55s

Provide feedback