Skip to content

Pull requests: google/qwix

New pull request New
23 Open 291 Closed
23 Open 291 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

Add fused Hadamard quantization Pallas kernel.
#321 opened Jun 29, 2026 by copybara-service Bot Loading…
Force int64 bias for w8a16
#320 opened Jun 29, 2026 by copybara-service Bot Loading…
Adding test
#319 opened Jun 24, 2026 by copybara-service Bot Loading…
[QwixODML] Add generalized binary Einsum flattening for ODML conversion
#315 opened Jun 22, 2026 by copybara-service Bot Loading…
Suppress new Pyrefly findings before upgrading to V1.1.1
#310 opened Jun 22, 2026 by copybara-service Bot Loading…
[QwixODML] Generalize 3D einsum flattening
#307 opened Jun 17, 2026 by copybara-service Bot Loading…
Adding support for int16 quantization.
#306 opened Jun 17, 2026 by copybara-service Bot Loading…
Introduce HiQArray and its HiJax integration.
#293 opened Jun 2, 2026 by copybara-service Bot Loading…
Internal
#290 opened Jun 1, 2026 by copybara-service Bot Loading…
Rename save/load_pytree to save/load. Eliminate most user-facing "pytree" terminology in favor of "state" as a more specific term. Add deprecations.py for handling deprecated public functions.
#280 opened May 18, 2026 by copybara-service Bot Loading…
[QwixODML] Support 3D RHS weights with middle contracting dimensions in Qwix ODML quantization.
#278 opened May 11, 2026 by copybara-service Bot Loading…
Add Sparsity NNX module
#236 opened Mar 27, 2026 by BirdsOfAFthr Contributor Loading…
2 tasks done
Internal
#232 opened Mar 26, 2026 by copybara-service Bot Loading…
Just updated the README.md
#209 opened Feb 5, 2026 by lshariprasad Loading…
1 of 2 tasks
[Tunix] Add _put_prompts_to_queue to handle dataset iteration and partial batches.
#202 opened Jan 29, 2026 by copybara-service Bot Loading…
Remove QuantizedArray class.
#180 opened Nov 24, 2025 by copybara-service Bot Loading…
Remove support for QuantizedArray inputs to attention.
#179 opened Nov 24, 2025 by copybara-service Bot Loading…
Remove the ptq change, move to another cl or use model surgery
#156 opened Nov 7, 2025 by copybara-service Bot Loading…
1
Add clip_gradients option to Qt quantization.
#132 opened Oct 27, 2025 by copybara-service Bot Loading…
Allow gradients on QArray
#131 opened Oct 27, 2025 by copybara-service Bot Loading…
qwix ragged dot in moe.py
#108 opened Sep 26, 2025 by copybara-service Bot Loading…
1
[TokamaxSplashAttentionQuantization] Quantize tokamax splash attention kernel forward pass using Qwix.
#97 opened Sep 18, 2025 by copybara-service Bot Loading…
Use native int2 types.
#7 opened Apr 30, 2025 by copybara-service Bot Loading…
ProTip! Mix and match filters to narrow down what you’re looking for.