jax

faf677af - Split the "multiaccelerator" tag into "multiaccelerator" and "multiaccelerator-only".

Split the "multiaccelerator" tag into "multiaccelerator" and "multiaccelerator-only". Some tests require multiple accelerators, whereas some just benefit from them. Use "multiaccelerator" for those that benefit, and "multiaccelerator-only" for those that require. For consistency, rename the pytest mark "multiaccelerator-only". This change is intended to ensure tests like profiler_test.py run in our presubmits that only have one GPU. PiperOrigin-RevId: 935633309