Skip to content

Pull requests: google/aqt

New pull request New
69 Open 650 Closed
69 Open 650 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

jax.numpy.clip: update use of deprecated arguments.
#754 opened Jan 11, 2025 by copybara-service bot Loading…
add an option to broadcast random noises for stochastic rounding
#753 opened Dec 7, 2024 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…
Add quant function with gradients
#642 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…
ProTip! no:milestone will show everything without a milestone.