Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add TE section to Pax readme #200

Merged
merged 3 commits into from
Aug 29, 2023
Merged

Add TE section to Pax readme #200

merged 3 commits into from
Aug 29, 2023

Conversation

ashors1
Copy link
Contributor

@ashors1 ashors1 commented Aug 29, 2023

  • Updates Pax container to nvcr.io/nvidia/jax:23.08-paxml-py3
  • Updates Pax readme to add details on Transformer Engine support

@ashors1 ashors1 changed the title Add TE section to readme Add TE section to Pax readme Aug 29, 2023
@ashors1 ashors1 closed this Aug 29, 2023
@ashors1 ashors1 reopened this Aug 29, 2023
@ashors1 ashors1 merged commit 51f9ace into main Aug 29, 2023
3 checks passed
@ashors1 ashors1 deleted the pax-te-documentation branch August 29, 2023 16:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants