dev jaxpp for rocm #50
Draft
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Co-authored-by: Sean Lee <selee@nvidia.com> Co-authored-by: Hanfeng Chen <hanfengc@nvidia.com> Co-authored-by: Jonathan Dekhtiar <jdekhtiar@nvidia.com>
Co-authored-by: Anxhelo Xhebraj <axhebraj@nvidia.com> Co-authored-by: Hanfeng Chen <hanfengc@nvidia.com> GitOrigin-RevId: 61102accc53dbd97375ea816717204ded7cc34c4
- b45f4a89e8cc1ae301702657e3e245f5f155a0f2 by Sean Lee <selee@nvidia.com> - e117f23a5a8622a9f976a99055a5a2daeac813bf by Anxhelo Xhebraj <axhebraj@nvidia.com> - 4416609d1973645005d6dbda11ed3a7739a47c97 by Anxhelo Xhebraj <axhebraj@nvidia.com> - aad5a3f50d027e242ff52002d8a3c2ec24787960 by Anxhelo Xhebraj <axhebraj@nvidia.com> - 88d64acfa2ebf664cbbe5ad9cb9ccf528befdf9b by Sean Lee <selee@nvidia.com> - 49fbcbf70ba53af9a26744bf5656a996789ad615 by Anxhelo Xhebraj <axhebraj@nvidia.com> - 8280b5e897aa26c501373de74f20a9078f051cbc by Anxhelo Xhebraj <axhebraj@nvidia.com> - f00c26fca920939e3928ff6fcfd79aff85b88bc2 by Anxhelo Xhebraj <axhebraj@nvidia.com> - e6b80b613fe419fe71c89be61b72aa77346eda04 by Sean Lee <selee@nvidia.com> Co-authored-by: Anxhelo Xhebraj <axhebraj@nvidia.com> GitOrigin-RevId: b45f4a89e8cc1ae301702657e3e245f5f155a0f2
- 2df5d245b15bdea608169d4879d4bf0ea06525a3 by Anxhelo Xhebraj <axhebraj@nvidia.com> - 61fd828a7585f84d429cb85ae2b4ce9bce2aecc9 by Sean Lee <selee@nvidia.com> Co-authored-by: Anxhelo Xhebraj <axhebraj@nvidia.com> GitOrigin-RevId: 2df5d245b15bdea608169d4879d4bf0ea06525a3
Signed-off-by: Sean Lee <selee@nvidia.com>
- 3c305e5ecca380d8eb2a030a3119bded971b8420 by Anxhelo Xhebraj <axhebraj@nvidia.com> - f5582ff15b97bbda4696353100338d454b7f1ef3 by Sean Lee <selee@nvidia.com> - c8028c7bbb80137e1f27d9ff818a165b38bb459d by Anxhelo Xhebraj <axhebraj@nvidia.com> Co-authored-by: Anxhelo Xhebraj <axhebraj@nvidia.com> GitOrigin-RevId: 3c305e5ecca380d8eb2a030a3119bded971b8420
- c82e20311cbe5a7fd68b2168db12c258e6a319ca by Anxhelo Xhebraj <axhebraj@nvidia.com> - f983eab19a798b72be28590dcdfe413407588bb3 by Anxhelo Xhebraj <axhebraj@nvidia.com> - db970979207d141fb37283741e9953e092eb8b11 by Sean Lee <selee@nvidia.com> - 595fb96836e7b5fc51b36b5517014ff2aed5c7be by Anxhelo Xhebraj <axhebraj@nvidia.com> - 0e5bb33fea0a4e4fd651f5eff7bce34a7e2d88b2 by Anxhelo Xhebraj <axhebraj@nvidia.com> - 2558e7d8b99549b6c4b0fe88e70128771a1ccff2 by Anxhelo Xhebraj <axhebraj@nvidia.com> Co-authored-by: Anxhelo Xhebraj <axhebraj@nvidia.com> GitOrigin-RevId: c82e20311cbe5a7fd68b2168db12c258e6a319ca
Signed-off-by: Sean Lee <selee@nvidia.com>
i-chaochen
reviewed
Dec 17, 2025
|
|
||
| RUN uv pip install --no-cache-dir -e '/workdir/jaxpp[dev]' | ||
| RUN uv pip install --no-cache-dir -e /workdir/maxtext[cuda_12] --resolution=lowest && \ | ||
| if [[ -n "$JAX_INSTALL_URL" ]]; then uv pip install $JAX_INSTALL_URL; fi |
| @@ -0,0 +1,78 @@ | |||
| # Overview | |||
|
|
|||
| This repository is a fork of [MaxText](https://github.com/AI-Hypercomputer/maxtext) created for training with [JaxPP](https://github.com/NVIDIA/jaxpp). | |||
There was a problem hiding this comment.
let's rm this jaxpp.README.md for now
gulsumgudukbay
force-pushed
the
rocm-main
branch
2 times, most recently
from
ccc3b28 to
fb31255
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.