[JAX] Add a test using inputs with different device orders for a single colocated Python call

[JAX] Add a test using inputs with different device orders for a single colocated Python call #7374