ray icon indicating copy to clipboard operation
ray copied to clipboard

Published 20 hours ago • verified publisher icon ray-project

[Ray Air] adding the jax trainer

Open JiahaoYao opened this issue 2 years ago • 15 comments

Why are these changes needed?

  • [x] implement jax trainer
  • [x] adding the jax trainer example with the mnist dataset
  • [x] adding the unit test for jax
  • [x] adding ci test (buildkite) for jax trainer exclusively

Related issue number

this PR is the integration of the previous two PR:

  • https://github.com/ray-project/ray/pull/24895
  • https://github.com/ray-project/ray/pull/25385

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 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
    • [x] This PR is not tested :(

JiahaoYao avatar Jul 27 '22 18:07 JiahaoYao

seemingly working here? is this real?

JiahaoYao avatar Jul 27 '22 22:07 JiahaoYao

^ the test passed locally

JiahaoYao avatar Aug 01 '22 20:08 JiahaoYao

@matthewdeng the pr is ready for review!

JiahaoYao avatar Aug 11 '22 23:08 JiahaoYao

for people who get the notification, please ignore this. i was merging the master

git fetch upstream
git merge upstream/master

and it sends the request for all the people.

JiahaoYao avatar Aug 21 '22 20:08 JiahaoYao

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Sep 30 '22 18:09 stale[bot]

keep alive

jiaodong avatar Sep 30 '22 18:09 jiaodong

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Nov 01 '22 12:11 stale[bot]

keep alive

JiahaoYao avatar Nov 01 '22 23:11 JiahaoYao

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Dec 03 '22 04:12 stale[bot]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Jan 02 '23 08:01 stale[bot]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Feb 02 '23 02:02 stale[bot]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Mar 11 '23 12:03 stale[bot]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Apr 25 '23 20:04 stale[bot]

keep alive

JiahaoYao avatar Apr 27 '23 06:04 JiahaoYao

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Jun 10 '23 05:06 stale[bot]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

  • If you'd like to keep this open, just leave any comment, and the stale label will be removed.

stale[bot] avatar Jul 15 '23 02:07 stale[bot]

Hi again! The issue will be closed because there has been no more activity in the 14 days since the last message.

Please feel free to reopen or open a new issue if you'd still like it to be addressed.

Again, you can always ask for help on our discussion forum or Ray's public slack channel.

Thanks again for opening the issue!

stale[bot] avatar Aug 11 '23 22:08 stale[bot]