jax

034c430c - Add support of executing jax tests from withing a parent workspace as @jax//tests:gpu_tests