Skip to content

Pull requests: google/aqt

New pull request New
64 Open 679 Closed
64 Open 679 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Handle the jax.promote_dtype failure for [int4, fp8_e4m3, fp8_e5m2] when using aqt_einsum
#759 opened Mar 4, 2025 by copybara-service bot Loading…
Update references to JAX's GitHub repo
#752 opened Dec 5, 2024 by copybara-service bot Loading…
dummy
#751 opened Nov 21, 2024 by copybara-service bot Loading…
Add rounding bias that controls the rounding threshold
#750 opened Nov 20, 2024 by copybara-service bot Loading…
Enable fake quant in the bwd with local_aqt
#748 opened Nov 13, 2024 by copybara-service bot Loading…
Update the accumulation dtype if FP8 precision is used for AQT.
#745 opened Nov 5, 2024 by copybara-service bot Loading…
[sharding_in_types][Take 2] Add out_type argument to einsum and dot_general to allow specifying for the output type. Right now, it only accept a NamedSharding but in the future we can allow a polymorphic type of: jax.ShapeDtypeStruct | Sharding | Layout.
#743 opened Oct 22, 2024 by copybara-service bot Loading…
cfg as gradient experiment - unsuccessful
#742 opened Oct 18, 2024 by anfals Loading…
Add prototype delayed scaling with overwrite with gradients
#741 opened Oct 17, 2024 by anfals • Draft
Fix dequant logic with zero scaling factors
#730 opened Sep 26, 2024 by copybara-service bot Loading…
Allows setting array of bounds in set_calibration_config.
#729 opened Sep 24, 2024 by copybara-service bot Loading…
QTensor refactoring: holds AqtTileMap instead of TilingState.
#692 opened Aug 9, 2024 by copybara-service bot Loading…
Add support asymmetric fake-quantization to AQTv2.
#675 opened Jul 23, 2024 by copybara-service bot Loading…
internal
#672 opened Jul 12, 2024 by copybara-service bot Loading…
Set config.set_bits params' default values as None.
#659 opened Jun 25, 2024 by copybara-service bot Loading…
Internal prototype
#655 opened Jun 21, 2024 by copybara-service bot Loading…
Change a way to split tiling axes without configuration
#644 opened Jun 7, 2024 by copybara-service bot Loading…
[EXPERIMENTAL] Remove leading 1s from the scale.
#643 opened Jun 6, 2024 by copybara-service bot Loading…
Support pl.BlockSpec that holds block_shape with None.
#630 opened May 24, 2024 by copybara-service bot Loading…
Add from_qtensor which transform QTensor into PallasQTensor.
#629 opened May 24, 2024 by copybara-service bot Loading…
Move PallasQTensor materialization code as a method. Call materialization before dequant. This simplifies user interface.
#628 opened May 24, 2024 by copybara-service bot Loading…
cast scale to dequant_type when restoring scale to original shape in pallas.
#627 opened May 24, 2024 by copybara-service bot Loading…
[Pallas Support] Add dot_general for pallas.
#624 opened May 22, 2024 by copybara-service bot Loading…
Put numerics and numerics-related logic into QTensor
#623 opened May 22, 2024 by copybara-service bot Loading…
[Pallas support] Add quant_blockwisely function, which quantizes tensor according to the BlockSpec of the inputs.
#596 opened Apr 24, 2024 by copybara-service bot Loading…
ProTip! What’s not been updated in a month: updated:<2025-06-03.