-
Notifications
You must be signed in to change notification settings - Fork 42
Issues: google/praxis
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
[Feature Request] Need Matmul Attention layer instead of Einsum to support GPU running
#46
opened Jan 27, 2024 by
MoFHeka
[Feature Request] Need ZeRo-1/2 to cooperate with PP+TP+DP. Which may more faster than FSDP sometimes.
#45
opened Jan 26, 2024 by
MoFHeka
gpu_fast_attention not passing segment_ids to jax pallas attention mha
#30
opened Oct 2, 2023 by
Cjkkkk
ProTip!
Follow long discussions with comments:>50.