refactor: unify learning rate schedulers with array API #13598

Job	Run time
Test Python (3, 3.13)	19m 6s
Test Python (7, 3.13)	23m 35s
Test Python (4, 3.10)	18m 10s
Test Python (6, 3.13)	23m 34s
Test Python (1, 3.10)	22m 24s
Test Python (10, 3.13)	18m 28s
Test Python (8, 3.10)	20m 1s
Test Python (3, 3.10)	23m 34s
Test Python (9, 3.10)	21m 52s
Test Python (2, 3.13)	21m 37s
Test Python (5, 3.13)	23m 34s
Test Python (5, 3.10)	18m 44s
Test Python (6, 3.10)	18m 9s
Test Python (8, 3.13)	20m 31s
Test Python (12, 3.13)	23m 34s
Test Python (11, 3.13)	20m 56s
Test Python (10, 3.10)	19m 12s
Test Python (7, 3.10)	18m 39s
Test Python (1, 3.13)	18m 7s
Test Python (12, 3.10)	20m 2s
Test Python (2, 3.10)	19m 14s
Test Python (9, 3.13)	18m 45s
Test Python (11, 3.10)	17m 50s
Test Python (4, 3.13)	18m 4s
Pass testing Python	2s
Combine and update integration test durations	1s
	8h 7m 45s

Provide feedback