feat(jax): force & virial #9434

Job	Run time
Test Python (1, 3.9)	28m 31s
Test Python (1, 3.12)	30m 0s
Test Python (2, 3.9)	27m 14s
Test Python (2, 3.12)	32m 4s
Test Python (3, 3.9)	27m 4s
Test Python (3, 3.12)	30m 11s
Test Python (4, 3.9)	27m 32s
Test Python (4, 3.12)	34m 36s
Test Python (5, 3.9)	26m 42s
Test Python (5, 3.12)	32m 23s
Test Python (6, 3.9)	28m 19s
Test Python (6, 3.12)	32m 44s
Combine and update integration test durations (3.9)	1m 27s
Combine and update integration test durations (3.12)	3m 23s
Pass testing Python	4s
	6h 2m 14s

Provide feedback