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
Prevent GPTQ from accidentally being "activated" during TRAIN mode
#662
copybara-service[bot]
closed
2 months ago
0
better SR
#661
copybara-service[bot]
closed
2 months ago
0
Version update for new AQT release.
#660
copybara-service[bot]
closed
2 months ago
0
Set config.set_bits params' default values as None.
#659
copybara-service[bot]
opened
2 months ago
0
Add an optional Quantizer field for scale dtype.
#658
copybara-service[bot]
closed
2 months ago
0
add fp numerics import in config
#657
copybara-service[bot]
closed
2 months ago
0
General floating point numerics
#656
copybara-service[bot]
closed
2 months ago
0
Internal prototype
#655
copybara-service[bot]
opened
2 months ago
0
Optionally tile axes in tiled_dot_general
#654
copybara-service[bot]
closed
2 months ago
0
Allow calibrate to return the updated lhs and rhs
#653
copybara-service[bot]
closed
2 months ago
0
Separate a function to apply use_fwd_quant flag.
#652
copybara-service[bot]
closed
2 months 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
Remove usage of xla_computation and replace with jax.jit(f).lower(*args).compile_ir()
#646
copybara-service[bot]
closed
3 months ago
0
Support auto sharding for subchannel on AQT level
#645
copybara-service[bot]
closed
2 months ago
0
Change a way to split tiling axes without configuration
#644
copybara-service[bot]
opened
3 months ago
0
[EXPERIMENTAL] Remove leading 1s from the scale.
#643
copybara-service[bot]
opened
3 months ago
0
Add quant function with gradients
#642
copybara-service[bot]
opened
3 months ago
0
Pull out local_aqt from the aqt_dot_general's __call__ function
#641
copybara-service[bot]
closed
3 months ago
0
Factor out tiling operations out of `tiled_dot_general`
#640
copybara-service[bot]
closed
3 months ago
0
Fix FP8 target training loss
#639
copybara-service[bot]
closed
3 months ago
0
Make spatial_dimensions a required field and clarify in docstring
#638
copybara-service[bot]
closed
3 months ago
0
New examples showcasing AQT usage for various quantization schemes.
#637
copybara-service[bot]
closed
2 months ago
0
Do not submit; just for internal review
#636
copybara-service[bot]
closed
3 months ago
0
Update version to 0.7.4
#635
copybara-service[bot]
closed
3 months ago
0
Added some TODOs
#634
copybara-service[bot]
closed
3 months ago
0
Make sure QTensor#dtype() returns an instance of jnp.dtype
#633
copybara-service[bot]
closed
3 months ago
0
A `transpose` function that permute order of axis through reshaping if possible.
#632
copybara-service[bot]
closed
3 months ago
0
A `transpose` function that permute order of axis through reshaping if possible.
#631
copybara-service[bot]
closed
3 months ago
0
Support pl.BlockSpec that holds block_shape with None.
#630
copybara-service[bot]
opened
3 months ago
0
Add `from_qtensor` which transform `QTensor` into `PallasQTensor`.
#629
copybara-service[bot]
opened
3 months ago
0
Move `PallasQTensor` materialization code as a method. Call materialization before `dequant`. This simplifies user interface.
#628
copybara-service[bot]
opened
3 months ago
0
cast scale to dequant_type when restoring scale to original shape in pallas.
#627
copybara-service[bot]
opened
3 months ago
0
A `transpose` function that permute order of axis through reshaping if possible.
#626
copybara-service[bot]
closed
3 months ago
0
Let the user provide custom tiling function to AqtEinsum and AqtDotGeneral
#625
copybara-service[bot]
closed
3 months ago
0
[Pallas Support] Add dot_general for pallas.
#624
copybara-service[bot]
opened
3 months ago
0
Put numerics and numerics-related logic into QTensor
#623
copybara-service[bot]
opened
3 months ago
0
internal change
#622
copybara-service[bot]
closed
3 months ago
0
Tiling simplifications
#621
copybara-service[bot]
closed
3 months ago
0
Improve the numerical stability by filtering out zero scaling factors
#620
copybara-service[bot]
closed
3 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
3 months ago
0
Replace deprecated `jax.tree_*` functions with `jax.tree.*`
#617
copybara-service[bot]
closed
3 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
Previous
Next