Skip to content

Pull requests: google/qwix

New pull request New
31 Open 293 Closed
31 Open 293 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 test for einsum with 2D scales.
#332 opened Jul 10, 2026 by copybara-service Bot Loading…
docs(mxfp): correct the outdated scaled_matmul TPU comment
#331 opened Jul 10, 2026 by lokic233 Loading…
feat(int4): add nibble_pack/unpack for true 0.5 B/elem int4 storage
#329 opened Jul 8, 2026 by lokic233 Loading…
[QwixODML] Add IS_UNTRANSFORMED_WEIGHT metadata key to prevent layout mismatch in per-channel quantization
#327 opened Jul 7, 2026 by copybara-service Bot Loading…
[QwixODML] Add generalized Einsum flattening for ODML conversion to support high-rank weights
#326 opened Jul 7, 2026 by copybara-service Bot Loading…
[QwixODML] Forward metadata during weight and output reshaping in flatten_dot_general
#325 opened Jul 7, 2026 by copybara-service Bot Loading…
docs: document MXFP numerics (mxfp8/mxfp4/nvfp4) and hardware dispatch
#324 opened Jul 6, 2026 by lokic233 Loading…
1
docs(mxfp): clarify that TPU/CPU use native fp8 dot_general fallback
#323 opened Jul 6, 2026 by lokic233 Loading…
1
[pallas:mosaic] Migrated all uses of pltpu.emit_pipeline to explicit indices
#322 opened Jul 3, 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…
ProTip! Exclude everything labeled bug with -label:bug.