Skip to content

Tags: google/qwix

Tags

0.1.5

Toggle 0.1.5's commit message
Merge pull request #186 from google:web

PiperOrigin-RevId: 842818044

0.1.4

Toggle 0.1.4's commit message
Rename padded_qarray to padded_ptq because it's more than qarray.

PiperOrigin-RevId: 831970069

0.1.3

Toggle 0.1.3's commit message
Bump version to 0.1.3

Also ensures Qwix works with Flax 0.12.0

PiperOrigin-RevId: 831559839

0.1.2

Toggle 0.1.2's commit message
Bump version to 0.1.2.

PiperOrigin-RevId: 825624343

0.1.1

Toggle 0.1.1's commit message
Switch to an alternative way to discover weight_name

This change introduces `flax_util.find_param` to dynamically find the name of a parameter within the current Flax/NNX module. This eliminates the need to store and propagate `weight_name` using `aux_data`, simplifying the interception logic in PTQ and LoRA providers. The `WithAux` class no longer stores `weight_name`.

The possible problem with this approach is we need some heuristics to find the weight name, and could possibly make mistake when the activation has the same shape as the weight, or there are multiple params with the same shape. In reality this should be rare and we have added proper checks for them.

PiperOrigin-RevId: 814293120

0.1.0

Toggle 0.1.0's commit message
bump the version to 0.1.0 for our first PyPI release.

PiperOrigin-RevId: 809201447