merge_array_jax Merges two sorted arrays into a single sorted array in Jax Benchmark: Parallel merge took 0.000411 seconds. Merge-then-sort took 0.000853 seconds.