Skip to content

Pull requests: google/praxis

New pull request New
15 Open 56 Closed
15 Open 56 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
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

[NVIDIA] Support GQA with jax.nn.dot_product_attention pull ready
#85 opened Aug 22, 2024 by kaixih Loading…
Use TE dpa for grok mqa
#84 opened Aug 20, 2024 by hx89 Loading…
Bump tensorflow from 2.9.3 to 2.12.1 in /praxis/pip_package in the pip group across 1 directory dependencies Pull requests that update a dependency file
#82 opened Aug 2, 2024 by dependabot bot Loading…
Fix checkpoint policy for Grok
#76 opened Jun 18, 2024 by hx89 Loading…
[draft] Fp8 inference experimental
#72 opened Jun 5, 2024 by wenscarl Loading…
Support fp8 direct quantization pull ready
#69 opened May 22, 2024 by wenscarl Loading…
3
Fix typo
#59 opened Apr 22, 2024 by ppwwyyxx Loading…
Add offloading checkpoint policy
#51 opened Mar 12, 2024 by jaro-sevcik Loading…
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.