feat(jax): zbl #9716

Job	Run time
Test Python (1, 3.9)	30m 48s
Test Python (1, 3.12)	41m 6s
Test Python (2, 3.9)	30m 17s
Test Python (2, 3.12)	41m 14s
Test Python (3, 3.9)	29m 33s
Test Python (3, 3.12)	41m 17s
Test Python (4, 3.9)	29m 54s
Test Python (4, 3.12)	40m 17s
Test Python (5, 3.9)	30m 39s
Test Python (5, 3.12)	40m 15s
Test Python (6, 3.9)	30m 39s
Test Python (6, 3.12)	43m 7s
Combine and update integration test durations (3.9)	2m 9s
Combine and update integration test durations (3.12)	6m 24s
Pass testing Python	1s
	7h 17m 40s

Provide feedback