Ecosyste.ms: Issues

An open API service for providing issue and pull request metadata for open source projects.

GitHub / google/aqt issues and pull requests

#601 - More tiled dot general tests but only on shapes

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago

#600 - generalized einsum or matmul api for pure jax

Issue - State: open - Opened by sh0416 7 months ago

#599 - NormalFloat4 support

Issue - State: open - Opened by sh0416 7 months ago

#598 - Remove int4 to int8 conversion in AQT

Pull Request - State: open - Opened by copybara-service[bot] 7 months ago

#597 - Add `quant` function, which quantizes tensor channel-wisely and returns QTensor and the blockspec of QTensor.

Pull Request - State: open - Opened by copybara-service[bot] 7 months ago

#596 - [Pallas support] Add `quant_blockwisely` function, which quantizes tensor according to the BlockSpec of the inputs.

Pull Request - State: open - Opened by copybara-service[bot] 7 months ago

#595 - How to use it with jnp.einsum?

Issue - State: open - Opened by sh0416 7 months ago - 1 comment

#594 - Bump the version number for a new release

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago

#593 - Migrate sparsity dependencies to new open source library.

Pull Request - State: open - Opened by copybara-service[bot] 7 months ago

#592 - Add transpose function to recover scale from scale_t.

Pull Request - State: closed - Opened by copybara-service[bot] 7 months ago

#591 - Remove the hardcoded use_fwd_quant=False in config_v4

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#590 - Set the default value for FreezerMode to NONE.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#589 - Regression test on binary pytree stability.

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#588 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#587 - Rename QTensor to QArray

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#586 - Add sparsity core library which will eventually replace sparsity_core_depr.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#585 - Adding new base directory for the open source sparsity library.

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#584 - Migrate AQTv1 calibration to AQTv2

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#583 - Put numerics and numerics-related logic into QTensor

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#582 - Add length function to QTensor

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#581 - Move cpu int4 workaround to better location

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#580 - Add sparsity core library which will eventually replace sparsity_core_depr.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#579 - Embedded inference or tflite compatibility on the roadmap?

Issue - State: open - Opened by soren-halter 8 months ago

#578 - Bump the version number

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#577 - Bump the version number

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#576 - Extend floating-point stochastic rounding to FP32

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#575 - Allow config_v4 and set_stochastic_rounding to work with FP8 numerics

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#574 - Merge sparsity libraries.

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#573 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#572 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#571 - Remove qvalue_astype on make_conv_general_dilated to accelerate of conv

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#570 - Deprecate / Remove aqt.make_dot_general

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#569 - inline assert_config_validity

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#568 - inline _default_dg_quantizer_make.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#567 - Replace Sequence[int] type to Sequence[AxisIdx] type whenever possible.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#566 - Add GPTQ as an extension of the AQT library.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#565 - SR for FP

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#564 - Enable Freezer for static range quantization.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#563 - Move the transpose functions to separate file.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#562 - Moves sparsity functionalities to sparsity folder.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#561 - Adding sparsity functionality for computing channelwise masks.

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#560 - Allow to inject custom DG into AQT

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#559 - inline effective numerics in make_quantizer

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#558 - Disallow passing None in make_conv_general_dilated()'s cfg param.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#557 - inline effective numerics in make_quantizer

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#556 - Move quantizer_make and its dependent function to aqt_quantizer

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#555 - Enable aqt_flax.py to set the custom wrapper for QTensor-saving Freezer.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#554 - Enable custom initialization wrapper for the Freezer.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#553 - Update Gemax to add an option to use the QTensor-saving Freezer.

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#552 - Add an option to not to store scale_t.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#551 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#550 - Update e2e test to use the QTensor-saving Freezer.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#549 - change the tile size in flax_e2e_example to 5 so that the shape change is more visible

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#548 - Binary quantization?

Issue - State: open - Opened by kishorenc 8 months ago

#547 - Easy configuration for einsum sub-channel quantization.

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#546 - This CL adds ridge regression as a method for post-processing the scales, reducing quantization error.

Pull Request - State: open - Opened by copybara-service[bot] 8 months ago

#545 - Fix the flakiness of //third_party/py/aqt/jax/v2/numerics:fp8_numerics_test_cpu

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#544 - Internal

Pull Request - State: closed - Opened by copybara-service[bot] 8 months ago

#543 - Add an option to choose between flax.nn.custom_vjp and jax.custom_vjp.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#542 - Adds a tutorial for how to make a basic AQT quantizer

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#541 - For AQT `intercept_method` API, Adds skeleton code and prototype for nn.Dense.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#540 - Replace jax.core.symbolic_equal_dim with ==

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#539 - Freezer implementation saving the whole QTensor, instead of saving its members separately

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#538 - Set the QTensor's dequant_dtype during SERVE mode to scale_t's dtype.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#537 - Add vlog to print out tensor dimensions before and after tiling.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#536 - Fix Presubmit failure after JAX update

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#535 - 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.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#534 - improve assertion message in tiled_dg

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#533 - Use sub-channel quantization for PAX AqtEinsum.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#532 - bump version number

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#531 - Internal.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#530 - Enable static range quantization. Focusing only for statistics collection.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#529 - Add command to flaky test, and sync the version number to the published one.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#528 - Fixed import error for `type.Self` for Python < 3.11

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#527 - Added pytype None checks

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#526 - bump up version number

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#525 - Refactor aqt_dot_general to run quantizer for both arguments at once

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#524 - Separate QuantMode as a stand-alone module.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#523 - Internal.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#522 - Adds a `allow_dummy_gradient_into_qtensor` toggle for turning off `quant_grad` sanity check under serving mode.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#521 - Enable qvalue to be None, and divide a function to calculate the scale and a function to calculate the actual qvalue inside quantizer.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#520 - Make quantizer produce values already cast to low-precision.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#519 - Use partial() to further simplify the integration of tiled_dot_general. This will be very similar to the tiling integration into pax ops.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#518 - Fix flaky fp8 test on CPU

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#517 - Remove unused import.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#516 - simplify the logic of tiled dot general in AqtDotGeneral. Results should be equivalent.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#515 - Fix cases of prng key reuse in aqt

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#514 - Can AQT be used to calculate qk score?

Issue - State: open - Opened by Lisennlp 9 months ago - 2 comments

#513 - test tiling dot general in flax e2e model

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#512 - Allow tiling configs to be more user friendly. Create a function that fills in the missing tiling specs and tiling axes in the config.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#511 - Partial integration of tiling into AqtDotGeneral class.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#510 - reduce the number of tests in tiled dot general test to fix flaky time out.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#509 - factor tiling process into a function

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#502 - TypeError: dataclass() got an unexpected keyword argument 'frozen'

Issue - State: closed - Opened by SetoKaiba 9 months ago - 2 comments

#495 - Make Freezer as a separated module, and make config to get it as input.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#492 - CALIBRATION PROTOTYPE #2: Extend Calibration as nn.Module, and inject it into config.

Pull Request - State: closed - Opened by copybara-service[bot] 9 months ago

#486 - PROTOTYPE CL FOR CALIBRATION

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#482 - jnp.int4 checkpointing work-around

Pull Request - State: closed - Opened by copybara-service[bot] 10 months ago

#364 - Synchronize the structure of context to that of config

Pull Request - State: open - Opened by copybara-service[bot] 12 months ago

#363 - Remove tf.cond from graph generation. The function will be replaced by multi-phase training.

Pull Request - State: open - Opened by copybara-service[bot] 12 months ago