issues
search
google
/
aqt
Apache License 2.0
248
stars
25
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Internal
#573
copybara-service[bot]
closed
5 months ago
0
Internal
#572
copybara-service[bot]
closed
5 months ago
0
Remove qvalue_astype on make_conv_general_dilated to accelerate of conv
#571
copybara-service[bot]
opened
5 months ago
0
Deprecate / Remove aqt.make_dot_general
#570
copybara-service[bot]
closed
5 months ago
0
inline assert_config_validity
#569
copybara-service[bot]
closed
5 months ago
0
inline _default_dg_quantizer_make.
#568
copybara-service[bot]
closed
5 months ago
0
Replace Sequence[int] type to Sequence[AxisIdx] type whenever possible.
#567
copybara-service[bot]
closed
5 months ago
0
Add GPTQ as an extension of the AQT library.
#566
copybara-service[bot]
closed
4 months ago
0
SR for FP
#565
copybara-service[bot]
closed
5 months ago
0
Enable Freezer for static range quantization.
#564
copybara-service[bot]
closed
5 months ago
0
Move the transpose functions to separate file.
#563
copybara-service[bot]
closed
5 months ago
0
Moves sparsity functionalities to sparsity folder.
#562
copybara-service[bot]
closed
5 months ago
0
Adding sparsity functionality for computing channelwise masks.
#561
copybara-service[bot]
opened
5 months ago
0
Allow to inject custom DG into AQT
#560
copybara-service[bot]
closed
5 months ago
0
inline effective numerics in make_quantizer
#559
copybara-service[bot]
closed
5 months ago
0
Disallow passing None in make_conv_general_dilated()'s cfg param.
#558
copybara-service[bot]
closed
5 months ago
0
inline effective numerics in make_quantizer
#557
copybara-service[bot]
closed
5 months ago
0
Move quantizer_make and its dependent function to aqt_quantizer
#556
copybara-service[bot]
closed
5 months ago
0
Enable aqt_flax.py to set the custom wrapper for QTensor-saving Freezer.
#555
copybara-service[bot]
closed
5 months ago
0
Enable custom initialization wrapper for the Freezer.
#554
copybara-service[bot]
closed
5 months ago
0
Update Gemax to add an option to use the QTensor-saving Freezer.
#553
copybara-service[bot]
opened
5 months ago
0
Add an option to not to store scale_t.
#552
copybara-service[bot]
closed
4 months ago
0
Internal
#551
copybara-service[bot]
closed
5 months ago
0
Update e2e test to use the QTensor-saving Freezer.
#550
copybara-service[bot]
closed
5 months ago
0
change the tile size in flax_e2e_example to 5 so that the shape change is more visible
#549
copybara-service[bot]
closed
5 months ago
0
Binary quantization?
#548
kishorenc
opened
5 months ago
0
Easy configuration for einsum sub-channel quantization.
#547
copybara-service[bot]
closed
6 months ago
0
This CL adds ridge regression as a method for post-processing the scales, reducing quantization error.
#546
copybara-service[bot]
opened
6 months ago
0
Fix the flakiness of //third_party/py/aqt/jax/v2/numerics:fp8_numerics_test_cpu
#545
copybara-service[bot]
closed
6 months ago
0
Internal
#544
copybara-service[bot]
closed
6 months ago
0
Add an option to choose between flax.nn.custom_vjp and jax.custom_vjp.
#543
copybara-service[bot]
closed
5 months ago
0
Adds a tutorial for how to make a basic AQT quantizer
#542
copybara-service[bot]
closed
6 months ago
0
For AQT `intercept_method` API, Adds skeleton code and prototype for nn.Dense.
#541
copybara-service[bot]
closed
5 months ago
0
Replace jax.core.symbolic_equal_dim with ==
#540
copybara-service[bot]
closed
6 months ago
0
Freezer implementation saving the whole QTensor, instead of saving its members separately
#539
copybara-service[bot]
closed
5 months ago
0
Set the QTensor's dequant_dtype during SERVE mode to scale_t's dtype.
#538
copybara-service[bot]
closed
6 months ago
0
Add vlog to print out tensor dimensions before and after tiling.
#537
copybara-service[bot]
closed
6 months ago
0
Fix Presubmit failure after JAX update
#536
copybara-service[bot]
closed
6 months ago
0
Extract tiled dg input shapes and tiled dimension numbers from jaxpr in order to correctly initialize the Freezer when tiled dg is enabled. The Freezer shape initialization of the injected aqt_dg will then match the tensor shape stored inside.
#535
copybara-service[bot]
closed
5 months ago
0
improve assertion message in tiled_dg
#534
copybara-service[bot]
closed
6 months ago
0
Use sub-channel quantization for PAX AqtEinsum.
#533
copybara-service[bot]
closed
6 months ago
0
bump version number
#532
copybara-service[bot]
closed
6 months ago
0
Internal.
#531
copybara-service[bot]
closed
6 months ago
0
Enable static range quantization. Focusing only for statistics collection.
#530
copybara-service[bot]
closed
5 months ago
0
Add command to flaky test, and sync the version number to the published one.
#529
copybara-service[bot]
closed
6 months ago
0
Fixed import error for `type.Self` for Python < 3.11
#528
copybara-service[bot]
closed
6 months ago
0
Added pytype None checks
#527
copybara-service[bot]
closed
6 months ago
0
bump up version number
#526
copybara-service[bot]
closed
6 months ago
0
Refactor aqt_dot_general to run quantizer for both arguments at once
#525
copybara-service[bot]
closed
6 months ago
0
Separate QuantMode as a stand-alone module.
#524
copybara-service[bot]
closed
6 months ago
0
Previous
Next