-
Notifications
You must be signed in to change notification settings - Fork 20
Pull requests: openxla/tokamax
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
[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…
Canonicalize attention precisions in a way that doesn't lose information.
#756
opened Mar 9, 2026 by
copybara-service
bot
Loading…
Remove obsolete
: ... from all python functions parameters annotations.
#755
opened Mar 7, 2026 by
copybara-service
bot
Loading…
Tokamax autotuning cache update. Auto-generated by xmanager/234249887
#714
opened Feb 27, 2026 by
copybara-service
bot
Loading…
Add
cuda-bench group of dependencies to pyproject.toml
#676
opened Feb 19, 2026 by
andportnoy
Loading…
Partition and parallelize numpy RNG when initializing large arrays.
#654
opened Feb 13, 2026 by
copybara-service
bot
Loading…
Add cuEquivariance triangle multiplication to Tokamax benchmarks.
#652
opened Feb 13, 2026 by
copybara-service
bot
Loading…
[pallas] Passing
backend= in addition to compiler_params= is redundant
#643
opened Feb 12, 2026 by
copybara-service
bot
Loading…
Upgrading pyproject dependencies for GPU to CUDA 13
#573
opened Jan 30, 2026 by
copybara-service
bot
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…
ProTip!
Mix and match filters to narrow down what you’re looking for.