Skip to content

Pull requests: google/praxis

New pull request New
14 Open 41 Closed
14 Open 41 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Support fp8 direct quantization
#69 opened May 22, 2024 by wenscarl Loading…
Enable dp in grok
#66 opened May 14, 2024 by hx89 Loading…
1
Fix Grok config
#64 opened May 6, 2024 by hx89 Loading…
1
Fix typo
#59 opened Apr 22, 2024 by ppwwyyxx Loading…
[NVIDIA] Add a custom layer for cudnn flash attention pull ready
#53 opened Mar 22, 2024 by kaixih Loading…
2
Add offloading checkpoint policy
#51 opened Mar 12, 2024 by jaro-sevcik Loading…
[NVIDIA] Use custom grad accumulation for FP8 params pull ready
#40 opened Jan 16, 2024 by kaixih Loading…
2
Add Transformer Engine support to Praxis
#36 opened Nov 28, 2023 by ashors1 Loading…
1
Added draft for grouped query attention
#34 opened Oct 27, 2023 by abhinavgoel95 • Draft
Praxis 1.2.0 release
#32 opened Oct 16, 2023 by chandrasekhard2 Loading…
1
Add alternate method to apply mask to allow XLA to detect MHA pattern
#27 opened Sep 19, 2023 by ashors1 Loading…
Bump certifi from 2023.5.7 to 2023.7.22 in /praxis/pip_package dependencies Pull requests that update a dependency file
#23 opened Jul 25, 2023 by dependabot bot Loading…
adding alternate method to apply mask to allow XLA to detect MHA pattern
#19 opened Jun 29, 2023 by abhinavgoel95 Loading…
1
Perform gradient clipping on global batch when using gradient accumulation pull ready
#6 opened Feb 14, 2023 by ashors1 Loading…
@zhangqiaorjc
9
ProTip! Adding no:label will show everything without a label.