Fix typo for JAX NeuronX naming

[devesr-amzn]

Description:

MANDATORY: PR needs test run output

Test Run Output:
N/A

Training tutorial:
N/A

Please make sure this PR contains correct classification terms (Alpha, Beta, and Stable).

If possible, provide your results or a link to them for the reviewer to check your work.

Issue #, sim, or t.corp if available:

Link to RTD for my changes:
https://awsdocs-neuron-staging.readthedocs-hosted.com/en/YOUR_BRANCH_NAME/

Additional context:

PR Checklist

• I've completely filled out the form above!
• (If applicable) I've automated a test to safegaurd my changes from regression.
• (If applicable) I've posted test collateral to prove my change was effective and not harmful.
• (If applicable) I've added someone from QA to the list of reviewers. Do this if you didn't make an automated test or feel it's appropriate for another reason.
• (If applicable) I've reviewed the licenses of updated and new binaries and their dependencies to make sure all licenses are on the pre-approved Amazon license list. See https://inside.amazon.com/en/services/legal/us/OpenSource/Pages/BlessedOpenSourceLicenses.aspx.

Pytest Marker Checklist

(Coming soon...)

Reviewer Checklist

• For reviewer, before merging, please cross-check:

• I've verified the changes render correctly on RTD (link above)
• I've ensured the submitter completed the form
• (If appropriate) I've run tests to verify the contents of the change

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.