Collect recursively and filter GPU tests using jax_test_gpu tag
#328
andportnoy
This run and associated checks have been archived and are scheduled for deletion.
Learn more about checks retention