Warp-JAX multi-GPU interoperability and added custom launch dimension #16

Job	Run time
build-and-test / build-warp-macos	1m 1s
build-and-test / build-warp-windows	7m 28s
build-and-test / build-warp-ubuntu	6m 36s
build-and-test / test-warp-macos	5m 24s
build-and-test / test-warp-ubuntu	18m 53s
build-and-test / test-warp-windows	13m 58s
	53m 20s

Provide feedback