Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Jun 10, 2026

• [QwixCore] Relax ragged_dot_qt_test thresholds for fp8_bwd on TPU.

  Show description for d8904fe
  

  guoren97

  authored and

  copybara-github

  committed
  d8904fe

  Copy full SHA for d8904fe

Commits on Jun 8, 2026

• [QwixODMLCoverage]Update QAT MAE threshold in ODML coverage test

  Show description for c2548f0
  

  guoren97

  authored and

  copybara-github

  committed
  c2548f0

  Copy full SHA for c2548f0

• [QwixMxfpNumerics] Implement MXFP/NVFP support in Qwix -- hardware acceleration for eligible devices and emulation for legacy devices.

  Show description for 83dec5f
  

  jshin1394

  authored and

  copybara-github

  committed
  83dec5f

  Copy full SHA for 83dec5f

Commits on Jun 5, 2026

• Add support for loading pre-quantized checkpoints into QT models for continued training.

  Show description for 5f33aad
  

  christinetung

  authored and

  copybara-github

  committed
  5f33aad

  Copy full SHA for 5f33aad

• [QwixStochasticRounding] Refactor noise functions from closures to a custom PyTree class (`NoiseFn`) in `stochastic_rounding.py`. Register `DotGeneralQtConfig` as a PyTree in `dot_general_qt.py`.

  Show description for 6daf86b
  

  jshin1394

  authored and

  copybara-github

  committed
  6daf86b

  Copy full SHA for 6daf86b

Commits on Jun 2, 2026

• Remove testcase with scoped vmem oom.

  Show description for 50e9511
  

  chapman20j

  authored and

  copybara-github

  committed
  50e9511

  Copy full SHA for 50e9511

• Refactor flax_util into common utilities folder.

  Show description for 5f70f5b
  

  christinetung

  authored and

  copybara-github

  committed
  5f70f5b

  Copy full SHA for 5f70f5b

Commits on Jun 1, 2026

• Add initial structure for HiJAX integration in qwix.

  Show description for 1288bdc
  

  chapman20j

  authored and

  copybara-github

  committed
  1288bdc

  Copy full SHA for 1288bdc

• Refactor process_prequantized_params to a new checkpoint utility module.

  Show description for cdfcc10
  

  christinetung

  authored and

  copybara-github

  committed
  cdfcc10

  Copy full SHA for cdfcc10

• Refactor to improve type safety and clarity in `qwix/_src/core`

  Show description for f056c3d
  

  chapman20j

  authored and

  copybara-github

  committed
  f056c3d

  Copy full SHA for f056c3d

Commits on May 29, 2026

• Generalize quantized matmul kernel to use variable subblock sizes and handle trivial scale shapes.

  Show description for 03b4c1d
  

  chapman20j

  authored and

  copybara-github

  committed
  03b4c1d

  Copy full SHA for 03b4c1d

• Add LHS-fused quantized matmul Pallas kernel.

  Show description for d6e094c
  

  chapman20j

  authored and

  copybara-github

  committed
  d6e094c

  Copy full SHA for d6e094c

Commits on May 28, 2026

• Skip PR Contributor Agreement workflow for pull requests opened by copybara-service[bot].

  Show description for 7897b37
  

  christinetung

  authored and

  copybara-github

  committed
  7897b37

  Copy full SHA for 7897b37

Commits on May 27, 2026

• Update Case 1 to recognize both quantized dict and WithAux as prequantized template, and remove obsolete Case 2.

  Show description for 48a35e9
  

  christinetung

  authored and

  copybara-github

  committed
  48a35e9

  Copy full SHA for 48a35e9

Commits on May 26, 2026

• Introduce jax.numpy.asarray interception in PtqProvider.

  Show description for 99daf61
  

  qwix-copybara

  authored and

  copybara-github

  committed
  99daf61

  Copy full SHA for 99daf61

Commits on May 25, 2026

• Robust parameter identification for Qwix.

  Show description for 6a33d87
  

  qwix-copybara

  authored and

  copybara-github

  committed
  6a33d87

  Copy full SHA for 6a33d87

Commits on May 20, 2026

• Adjust renamed Orbax V1 API calls.

  Show description for 0c51b6e
  

  cpgaffney1

  authored and

  copybara-github

  committed
  0c51b6e

  Copy full SHA for 0c51b6e

Commits on May 19, 2026

• Merge pull request #279 from JyotinderSingh:ci

  Show description for 6dc08ad
  copybara-github

  committed
  6dc08ad

  Copy full SHA for 6dc08ad

Commits on May 18, 2026

• Update quantized matmul kernel to optimize performance for fp32.

  Show description for c47c82c
  

  chapman20j

  authored and

  copybara-github

  committed
  c47c82c

  Copy full SHA for c47c82c

• [CI] Fix workflow failures: use Python 3.11+, scope pyrefly to qwix/, skip google3 test

  Show description for 75e61d0
  JyotinderSingh

  committed
  75e61d0

  Copy full SHA for 75e61d0

Commits on May 13, 2026

• ci

  JyotinderSingh

  committed
  049b6a1

  Copy full SHA for 049b6a1

Commits on May 11, 2026

• pyrefly migration

  Show description for 8e968b1
  

  chapman20j

  authored and

  copybara-github

  committed
  8e968b1

  Copy full SHA for 8e968b1

Commits on May 6, 2026

• Add a parameterized test case for the `dot_general` kernel to test other shapes and blockspecs.

  Show description for 6b302f9
  

  chapman20j

  authored and

  copybara-github

  committed
  6b302f9

  Copy full SHA for 6b302f9

• [QwixCore] Strengthen scale safety check to handle subnormal scales on TPU

  Show description for e0dafa3
  

  qwix-copybara

  authored and

  copybara-github

  committed
  e0dafa3

  Copy full SHA for e0dafa3

Commits on May 5, 2026

• [QwixTAP] Fix Qwix tests after XLA simplification breakage.

  Show description for 893b2b0
  

  jshin1394

  authored and

  copybara-github

  committed
  893b2b0

  Copy full SHA for 893b2b0

Commits on May 4, 2026

• [QwixGptqNNX] Add GPTQ tests for Flax NNX models

  Show description for 1ce3dd2
  

  jshin1394

  authored and

  copybara-github

  committed
  1ce3dd2

  Copy full SHA for 1ce3dd2

Commits on Apr 28, 2026

• Improve error message for invalid quantization rule.

  Show description for 44d0f52
  

  JyotinderSingh

  authored and

  copybara-github

  committed
  44d0f52

  Copy full SHA for 44d0f52

Commits on Apr 23, 2026

• [QwixPrequantizedPallasLayer] Add explanatory comment for Case 2 in process_prequantized_params

  Show description for b966dc4
  

  jshin1394

  authored and

  copybara-github

  committed
  b966dc4

  Copy full SHA for b966dc4

Commits on Apr 22, 2026

• Add Smooth Quantization (SQ) to Qwix contrib.

  Show description for 0692041
  

  chapman20j

  authored and

  copybara-github

  committed
  0692041

  Copy full SHA for 0692041

• [QwixSubchannelBwd] Fix mathematical error when reusing tiled residuals in backward pass

  Show description for 2bc7aa5
  

  jshin1394

  authored and

  copybara-github

  committed
  2bc7aa5

  Copy full SHA for 2bc7aa5

Commits on Apr 21, 2026

• [QwixOdml] Enhance Qwix `odml.py` docstring and inline comments to enhance interpretability.

  Show description for b4828d2
  

  jshin1394

  authored and

  copybara-github

  committed
  b4828d2

  Copy full SHA for b4828d2

Commits on Apr 20, 2026

• [QwixQuantStat] Refactor `_collect_quant_stat` to `_update_and_get_quant_stat`.

  Show description for db2fee1
  

  jshin1394

  authored and

  copybara-github

  committed
  db2fee1

  Copy full SHA for db2fee1

• [QwixOdmlQuantizationBoundary] Fix MLIR assertion failure and metadata leakage in U-Net quantization.

  Show description for 3407075
  

  jshin1394

  authored and

  copybara-github

  committed
  3407075

  Copy full SHA for 3407075

Commits on Apr 18, 2026

• [QwixOdmlAuxData] Refactor ODML AuxData keys to use centralized Enum.

  Show description for 19eefb3
  

  jshin1394

  authored and

  copybara-github

  committed
  19eefb3

  Copy full SHA for 19eefb3

Commits on Apr 17, 2026

• Merge pull request #255 from BirdsOfAFthr:amandaliang

  Show description for dc38600
  copybara-github

  committed
  dc38600

  Copy full SHA for dc38600