jax

7bc8045f - Merge pull request #34871 from ezhulenev:device-groups