• upgrade for A3-High, A3-Mega, and a3-Ultra

Testing:

• A3-Mega: 5 consecutive successful runs of NeMo framework example
• A3-Ultra: 5 consecutive successful runs of NeMo framework example

Submission Checklist

NOTE: Community submissions can take up to 2 weeks to be reviewed.

Please take the following actions before submitting this pull request.

• Fork your PR branch from the Toolkit "develop" branch (not main)
• Test all changes with pre-commit in a local branch #
• Confirm that "make tests" passes all tests
• Add or modify unit tests to cover code changes
• Ensure that unit test coverage remains above 80%
• Update all applicable documentation
• Follow Cluster Toolkit Contribution guidelines #