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

Improve pjit guide #2755

Closed
IvyZX opened this issue Dec 23, 2022 · 0 comments · Fixed by #3016
Closed

Improve pjit guide #2755

IvyZX opened this issue Dec 23, 2022 · 0 comments · Fixed by #3016
Assignees
@IvyZX
Labels
Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.

Comments

@IvyZX
Copy link
Collaborator

IvyZX commented Dec 23, 2022

The pjit guide #2730 is out, and just to summarize a few items to visit in the future:

  • Change the pip3 install line when v0.6.4 is out.
  • Visit new JAX APIs like jax.sharding, device_put and jit after the APIs are more finalized.
  • Try out the use cases of pjit or jit without input/output sharding specifications.
@IvyZX IvyZX added the Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment. label Dec 23, 2022
@IvyZX IvyZX self-assigned this Dec 23, 2022
This was referenced Apr 5, 2023
Merged
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@IvyZX IvyZX

Labels
Priority: P2 - no schedule Best effort response and resolution. We have no plan to work on this at the moment.
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

Upgrade the pjit guide to jax.jit IvyZX/flax
1 participant
@IvyZX