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.

Sign up for GitHub

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

Jump to bottom

[cherry-pick] [docs] fine tune llama with trainium (#48768) #48854

Merged
merged 1 commit into from
Nov 22, 2024
Merged

[cherry-pick] [docs] fine tune llama with trainium (#48768) #48854

merged 1 commit into from
Nov 22, 2024

Conversation

chris-ray-zhang
Copy link
Contributor

Introduce a new Ray Train example for AWS Trainium.

CleanShot 2024-11-16 at 12 48 57@2x

Marked it as a community example as it is something we are collaborating with AWS Neuron team.

CleanShot 2024-11-16 at 12 48 37@2x

Docs screenshots

Screenshot 2024-11-20 at 11 19 39 AM Screenshot 2024-11-20 at 11 19 47 AM Screenshot 2024-11-20 at 11 19 54 AM

Why are these changes needed?

Related issue number

Checks

  • I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
  • I've run scripts/format.sh to lint the changes in this PR.
  • I've included any doc changes needed for https://docs.ray.io/en/master/.
    • I've added any new APIs to the API Reference. For example, if I added a
      method in Tune, I've added it in doc/source/tune/api/ under the
      corresponding .rst file.
  • I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
  • Testing Strategy
    • Unit tests
    • Release tests
    • This PR is not tested :(

@saihaj @chris-ray-zhang
docs: fine tune llama with trainium (ray-project#48768)
215db74 
Introduce a new Ray Train example for AWS Trainium. 

![CleanShot 2024-11-16 at 12 48
57@2x](https://github.com/user-attachments/assets/8b7d12d8-846f-497f-ba25-fd8a613f9007)

Marked it as a community example as it is something we are collaborating
with AWS Neuron team.

![CleanShot 2024-11-16 at 12 48
37@2x](https://github.com/user-attachments/assets/589d8ff3-fcb6-4b90-865d-006bcb4815a3)

Docs screenshots

<img width="1142" alt="Screenshot 2024-11-20 at 11 19 39 AM"
src="https://github.com/user-attachments/assets/aa3dadf7-96b9-46cc-8b6d-44c3e3bc3e1e">
<img width="1161" alt="Screenshot 2024-11-20 at 11 19 47 AM"
src="https://github.com/user-attachments/assets/859508fd-e47e-4758-a4c7-f15a749ece82">
<img width="1149" alt="Screenshot 2024-11-20 at 11 19 54 AM"
src="https://github.com/user-attachments/assets/28858f36-8cca-4eaa-a8ec-a1f7dda899d0">

---------

Signed-off-by: Saihajpreet Singh <[email protected]>
Co-authored-by: Chris Zhang <[email protected]>
@chris-ray-zhang chris-ray-zhang added v2.39.0 v2.39.0-pick release-blocker P0 Issue that blocks the release and removed v2.39.0 labels Nov 21, 2024
@angelinalg angelinalg changed the title docs: fine tune llama with trainium (#48768) [cherry-pick] [docs] fine tune llama with trainium (#48768) Nov 21, 2024
angelinalg
angelinalg approved these changes Nov 21, 2024
View reviewed changes
Copy link
Contributor

@angelinalg angelinalg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

stamp

@can-anyscale can-anyscale enabled auto-merge (squash) November 21, 2024 22:31
@github-actions github-actions bot added the go add ONLY when ready to merge, run all tests label Nov 21, 2024
@can-anyscale can-anyscale merged commit a46e3f4 into ray-project:releases/2.39.0 Nov 22, 2024
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@angelinalg angelinalg angelinalg approved these changes

@saihaj saihaj saihaj approved these changes

@hongpeng-guo hongpeng-guo Awaiting requested review from hongpeng-guo hongpeng-guo is a code owner

@justinvyu justinvyu Awaiting requested review from justinvyu justinvyu is a code owner

@matthewdeng matthewdeng Awaiting requested review from matthewdeng matthewdeng is a code owner

@raulchen raulchen Awaiting requested review from raulchen raulchen is a code owner

@woshiyyya woshiyyya Awaiting requested review from woshiyyya woshiyyya is a code owner

Assignees
No one assigned
Labels
go add ONLY when ready to merge, run all tests release-blocker P0 Issue that blocks the release v2.39.0-pick
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
4 participants
@chris-ray-zhang @saihaj @angelinalg @can-anyscale