Add support for `axis_name` and `axis_index_groups` to `lax.ragged_all_to_all` #25660

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

@ghpvnist

The following contributors were found for this pull request:

✅ 809e113 PR Opener: @copybara-service[bot]
✅ 809e113 Author: @ghpvnist <gu****n@google.com>

(Only the first commit for a unique contributor is listed.)