issues
search
google
/
aqt
Apache License 2.0
236
stars
25
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Comments for fp numerics
#696
copybara-service[bot]
closed
4 days ago
0
Use a different method name other than the variable name.
#695
copybara-service[bot]
closed
5 days ago
0
make dummy gradient into qtensor configurable
#694
copybara-service[bot]
closed
2 days ago
0
Adjust test mnist training loss because of changes in XLA
#693
copybara-service[bot]
closed
6 days ago
0
QTensor refactoring: holds AqtTileMap instead of TilingState.
#692
copybara-service[bot]
opened
1 week ago
0
Gradient quantization support
#691
lixcli
opened
1 week ago
0
Add a setter to enable int2 weight dtype.
#690
copybara-service[bot]
closed
5 days ago
0
cleaner FP numerics with subnormals supported
#689
copybara-service[bot]
opened
1 week ago
0
subnormal implementation with both static and stochastic rounding
#688
copybara-service[bot]
closed
2 days ago
0
Add support for Delayed Per-Tensor Scaling
#687
copybara-service[bot]
opened
1 week ago
0
Support subnormal in FP simulation
#686
copybara-service[bot]
opened
2 weeks ago
0
Allow space-to-batch propagations on Reduce
#685
copybara-service[bot]
closed
2 weeks ago
0
Consolidate the calculation of scale tensors into a function.
#684
copybara-service[bot]
opened
2 weeks ago
0
Support negative index on `AxisTiling` and `tiled_dot_general.to_tiled_axes_transposed`.
#683
copybara-service[bot]
closed
1 week ago
0
Add simple quantization range validation tests for `int_numerics`.
#682
copybara-service[bot]
closed
2 weeks ago
0
Add support for passing a dataset to `flax_e2e_model.train_and_evaluate` and make statistics printing easier to read.
#681
copybara-service[bot]
closed
2 weeks ago
0
Add fake-quantization support for quantization biases to AQTv2.
#680
copybara-service[bot]
closed
1 week ago
0
Remove intermediate `new_snr_highs` tensor.
#679
copybara-service[bot]
closed
2 weeks ago
0
[xla:cpu] Update expected target_loss for xla:cpu thunks
#678
copybara-service[bot]
closed
2 weeks ago
0
Modifying conv ops to optionally accept quantized lhs and rhs and also return them
#677
copybara-service[bot]
closed
4 days ago
0
[pallas] Relands PR #22552: Simplify handling of BlockMapping and GridMapping
#676
copybara-service[bot]
closed
3 weeks ago
0
Add support asymmetric fake-quantization to AQTv2.
#675
copybara-service[bot]
opened
3 weeks ago
0
Support subchannel quantization on `quant` and `calibrate`.
#674
copybara-service[bot]
closed
2 weeks ago
0
Bugfix on e2e example: fix wrong activation quantization mode.
#673
copybara-service[bot]
closed
1 month ago
0
internal
#672
copybara-service[bot]
opened
1 month ago
0
Add mean of abs calibration
#671
copybara-service[bot]
closed
1 month ago
0
Automatically selecting scales (clipping factors) based on SNR in AQT.
#670
copybara-service[bot]
closed
3 weeks ago
0
Updated pl.BlockSpec argument order following google/jax#22209
#669
copybara-service[bot]
closed
1 month ago
0
Open sourcing AQT's pallas API.
#668
copybara-service[bot]
closed
1 month ago
0
flax_e2e_model.py example fails
#667
Jconn
closed
1 week ago
0
QTensor refactoring: add TilingState on QTensor
#666
copybara-service[bot]
closed
3 weeks ago
0
Mean of abs calibration
#665
copybara-service[bot]
closed
1 week ago
0
dilated conv layer that can be injected into flax.nn.Conv
#664
Jconn
opened
1 month ago
0
Make local aqt support tiling the largest contraction axis (not the first) in the bwd.
#663
copybara-service[bot]
closed
1 month ago
0
Prevent GPTQ from accidentally being "activated" during TRAIN mode
#662
copybara-service[bot]
closed
1 month ago
0
better SR
#661
copybara-service[bot]
closed
1 month ago
0
Version update for new AQT release.
#660
copybara-service[bot]
closed
1 month ago
0
Set config.set_bits params' default values as None.
#659
copybara-service[bot]
opened
1 month ago
0
Add an optional Quantizer field for scale dtype.
#658
copybara-service[bot]
closed
1 month ago
0
add fp numerics import in config
#657
copybara-service[bot]
closed
1 month ago
0
General floating point numerics
#656
copybara-service[bot]
closed
1 month ago
0
Internal prototype
#655
copybara-service[bot]
opened
1 month ago
0
Optionally tile axes in tiled_dot_general
#654
copybara-service[bot]
closed
1 month ago
0
Allow calibrate to return the updated lhs and rhs
#653
copybara-service[bot]
closed
1 month ago
0
Separate a function to apply use_fwd_quant flag.
#652
copybara-service[bot]
closed
1 month ago
0
Disband DotGeneralRaw into dg_core_fwd and dg_core_bwd.
#651
copybara-service[bot]
opened
2 months ago
0
Separate dimension number update function for backward
#650
copybara-service[bot]
closed
2 months ago
0
Refactor: Separate quant() function from dot_general_raw.
#649
copybara-service[bot]
closed
2 months ago
0
Move calibration instantiation in fwd pass to outside of custom_vjp to allow a stateful bwd pass calibrators.
#648
copybara-service[bot]
closed
2 months ago
0
Modifying target loss because of XLA change.
#647
copybara-service[bot]
closed
2 months ago
0
Next