Skip to content

Pull requests: openxla/tokamax

New pull request New
6 Open 496 Closed
6 Open 496 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Explicitly pass input arguments as residuals to VJP function.
#490 opened Dec 16, 2025 by copybara-service bot Loading…
Add sm100 attention kernel.
#487 opened Dec 15, 2025 by copybara-service bot Loading…
test for hunting cuda error
#485 opened Dec 15, 2025 by copybara-service bot Loading…
Integrate Triton up to 8d445186
#456 opened Dec 9, 2025 by copybara-service bot Loading…
Factor out JAX SplashAttention into base.py.
#451 opened Dec 5, 2025 by copybara-service bot Loading…
Integrate Triton up to [8d445186](https://github.com/openai/triton/commits/ce7393e1e407a84513f6f133f93791878d445186)
#443 opened Dec 4, 2025 by copybara-service bot Loading…
ProTip! no:milestone will show everything without a milestone.