issues
search
google
/
aqt
Apache License 2.0
247
stars
25
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Improve the numerical stability by filtering out zero scaling factors
#620
copybara-service[bot]
closed
4 months ago
0
[XLA:SPACE_TO_BATCH] correctly propagate on dot
#619
copybara-service[bot]
closed
3 months ago
0
(pure refactoring) Simplify tiling code towards merging with QTensor
#618
copybara-service[bot]
closed
4 months ago
0
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#617
copybara-service[bot]
closed
4 months ago
0
[XLA:SPACE_TO_BATCH] correctly propagate on dot
#616
copybara-service[bot]
closed
4 months ago
0
Update version number
#615
copybara-service[bot]
closed
4 months ago
0
Remove None option from the container_dtype parameter in aqt_tensor.zeros_with_scale().
#614
copybara-service[bot]
closed
3 months ago
0
Remove the qdtype parameter from aqt_tensor.zeros_with_scale()
#613
copybara-service[bot]
closed
4 months ago
0
Internal - dirty prototype
#612
copybara-service[bot]
closed
3 months ago
0
Make AQT data classes use keyword-only fields (3/N)
#611
copybara-service[bot]
closed
4 months ago
0
Add scale_dtype argument to QTensor.zeros_with_scale.
#610
copybara-service[bot]
closed
4 months ago
0
Make AQT data classes to use keyword-only fields (2/N)
#609
copybara-service[bot]
closed
4 months ago
0
Add a util function to construct QTensor partition specs.
#608
copybara-service[bot]
closed
4 months ago
0
Internal
#607
copybara-service[bot]
closed
4 months ago
0
Fix flakiness in AQT tests
#606
copybara-service[bot]
closed
4 months ago
0
Use default values for Tensor dataclass rather than *_make() pattern
#605
copybara-service[bot]
closed
4 months ago
0
Make some AQT dataclasses to use keyword-only fields (1/N)
#604
copybara-service[bot]
closed
4 months ago
0
Set shardings of the scale to be the same as qvalue, except the contracting dimensions.
#603
copybara-service[bot]
closed
4 months ago
0
Add a dtype alias to QTensor. dequant_dtype should now be treated as private.
#602
copybara-service[bot]
closed
4 months ago
0
More tiled dot general tests but only on shapes
#601
copybara-service[bot]
closed
4 months ago
0
generalized einsum or matmul api for pure jax
#600
sh0416
opened
4 months ago
0
NormalFloat4 support
#599
sh0416
opened
4 months ago
0
Remove int4 to int8 conversion in AQT
#598
copybara-service[bot]
closed
4 months ago
0
Add `quant` function, which quantizes tensor channel-wisely and returns QTensor and the blockspec of QTensor.
#597
copybara-service[bot]
closed
4 months ago
0
[Pallas support] Add `quant_blockwisely` function, which quantizes tensor according to the BlockSpec of the inputs.
#596
copybara-service[bot]
opened
4 months ago
0
How to use it with jnp.einsum?
#595
sh0416
opened
4 months ago
1
Bump the version number for a new release
#594
copybara-service[bot]
closed
4 months ago
0
Migrate sparsity dependencies to new open source library.
#593
copybara-service[bot]
opened
4 months ago
0
Add transpose function to recover scale from scale_t.
#592
copybara-service[bot]
closed
4 months ago
0
Remove the hardcoded use_fwd_quant=False in config_v4
#591
copybara-service[bot]
closed
5 months ago
0
Set the default value for FreezerMode to NONE.
#590
copybara-service[bot]
closed
4 months ago
0
Regression test on binary pytree stability.
#589
copybara-service[bot]
opened
5 months ago
0
Internal
#588
copybara-service[bot]
closed
5 months ago
0
Rename QTensor to QArray
#587
copybara-service[bot]
opened
5 months ago
0
Add sparsity core library which will eventually replace sparsity_core_depr.
#586
copybara-service[bot]
closed
5 months ago
0
Adding new base directory for the open source sparsity library.
#585
copybara-service[bot]
opened
5 months ago
0
Migrate AQTv1 calibration to AQTv2
#584
copybara-service[bot]
closed
5 months ago
0
Put numerics and numerics-related logic into QTensor
#583
copybara-service[bot]
closed
2 months ago
0
Add length function to QTensor
#582
copybara-service[bot]
closed
5 months ago
0
Move cpu int4 workaround to better location
#581
copybara-service[bot]
closed
5 months ago
0
Add sparsity core library which will eventually replace sparsity_core_depr.
#580
copybara-service[bot]
closed
5 months ago
0
Embedded inference or tflite compatibility on the roadmap?
#579
soren-halter
opened
5 months ago
0
Bump the version number
#578
copybara-service[bot]
closed
5 months ago
0
Bump the version number
#577
copybara-service[bot]
closed
5 months ago
0
Extend floating-point stochastic rounding to FP32
#576
copybara-service[bot]
closed
5 months ago
0
Allow config_v4 and set_stochastic_rounding to work with FP8 numerics
#575
copybara-service[bot]
closed
5 months ago
0
Merge sparsity libraries.
#574
copybara-service[bot]
opened
5 months ago
0
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
Previous
Next