Collect recursively and filter GPU tests using `jax_test_gpu` tag#1091

olupton merged 5 commits intomainNVIDIA/JAX-Toolbox:mainfrom

aportnoy/use-jax_test_gpuNVIDIA/JAX-Toolbox:aportnoy/use-jax_test_gpuCopy head branch name to clipboard