Skip to content

Pull requests: openxla/tokamax

New pull request New
20 Open 845 Closed
20 Open 845 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

Bump minimum JAX version to 0.10.0.
#871 opened Apr 17, 2026 by copybara-service Bot Loading…
Avoid usage of deprecated jax.core APIs.
#860 opened Apr 15, 2026 by copybara-service Bot Loading…
Change benchmarking timers to run multiple iterations together.
#841 opened Apr 8, 2026 by stzeng Collaborator Loading…
Test
#838 opened Apr 8, 2026 by sbodenstein Collaborator Loading…
Clarify block_mask values in documentation
#804 opened Mar 27, 2026 by Guitaricet Loading…
1
Triton and Mosaic for linear_softmax_cross_entropy_loss
#801 opened Mar 27, 2026 by captainpete Loading…
fix: nested masks
#799 opened Mar 25, 2026 by emergenz Loading…
[Mosaic GPU] Remove the partitioned from MGPU APIs
#777 opened Mar 13, 2026 by copybara-service Bot Loading…
Reduce the number of arg specs tested under the ragged_dot API test.
#770 opened Mar 11, 2026 by copybara-service Bot Loading…
Add missing tcgen05.fence instructions to SM100 MGPU attention.
#757 opened Mar 9, 2026 by copybara-service Bot Loading…
Update triangle multiplication benchmark and pin JAX to >=0.9.0.1
#730 opened Mar 2, 2026 by rpachauri Collaborator • Draft
Add cuda-bench group of dependencies to pyproject.toml
#676 opened Feb 19, 2026 by andportnoy Member Loading…
@sbodenstein
Partition and parallelize numpy RNG when initializing large arrays.
#654 opened Feb 13, 2026 by copybara-service Bot Loading…
Add CuTeDSL Blackwell tutorial GEMM
#647 opened Feb 12, 2026 by andportnoy Member Loading…
34
[pallas] Passing backend= in addition to compiler_params= is redundant
#643 opened Feb 12, 2026 by copybara-service Bot Loading…
Add more tuned tiling
#612 opened Feb 7, 2026 by BirdsOfAFthr Contributor Loading…
Remove redundant bind call in Op.__call__.
#524 opened Jan 14, 2026 by copybara-service Bot Loading…
Raise an error on autotuning cache miss if configured.
#511 opened Jan 6, 2026 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…
ProTip! Filter pull requests by the default branch with base:main.