Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 2, 2024

• Allow space-to-batch propagations on Reduce

  Show description for cd60f00
  

  amitsabne1

  authored and

  copybara-github

  committed
  cd60f00

  Copy full SHA for cd60f00

Commits on Aug 1, 2024

• Add simple quantization range validation tests for int_numerics.

  Show description for 281cdd6
  

  phoenix-meadowlark

  authored and

  copybara-github

  committed
  281cdd6

  Copy full SHA for 281cdd6

• Support subchannel quantization on quant and calibrate.

  Show description for a9eeab5
  

  lenscloth

  authored and

  copybara-github

  committed
  a9eeab5

  Copy full SHA for a9eeab5

Commits on Jul 30, 2024

• Add support for passing a dataset to flax_e2e_model.train_and_evaluate and make statistics printing easier to read.

  Show description for 4cdb03d
  

  phoenix-meadowlark

  authored and

  copybara-github

  committed
  4cdb03d

  Copy full SHA for 4cdb03d

Commits on Jul 29, 2024

• Remove intermediate new_snr_highs tensor.

  Show description for 461159a
  

  dansuh17

  authored and

  copybara-github

  committed
  461159a

  Copy full SHA for 461159a

• [xla:cpu] Update expected target_loss for xla:cpu thunks

  Show description for 92aa2fe
  

  ezhulenev

  authored and

  copybara-github

  committed
  92aa2fe

  Copy full SHA for 92aa2fe

Commits on Jul 25, 2024

• [pallas] Relands PR #22552: Simplify handling of BlockMapping and GridMapping

  Show description for 8b68bc8
  

  gnecula

  authored and

  copybara-github

  committed
  8b68bc8

  Copy full SHA for 8b68bc8

• QTensor refactoring: add TilingState on QTensor

  Show description for 1db3f81
  

  lenscloth

  authored and

  copybara-github

  committed
  1db3f81

  Copy full SHA for 1db3f81

Commits on Jul 24, 2024

• Automatically selecting scales (clipping factors) based on SNR in AQT.

  Show description for fd70257
  

  dansuh17

  authored and

  copybara-github

  committed
  fd70257

  Copy full SHA for fd70257

Commits on Jul 18, 2024

• Bugfix on e2e example: fix wrong activation quantization mode.

  Show description for 5e5897c
  

  cdh4696

  authored and

  copybara-github

  committed
  5e5897c

  Copy full SHA for 5e5897c

Commits on Jul 9, 2024

• Updated pl.BlockSpec argument order following jax-ml/jax#22209

  Show description for 6ff5c0d
  

  superbobry

  authored and

  copybara-github

  committed
  6ff5c0d

  Copy full SHA for 6ff5c0d

• Open sourcing AQT"s pallas API.

  Show description for 273624c
  

  lenscloth

  authored and

  copybara-github

  committed
  273624c

  Copy full SHA for 273624c

Commits on Jun 28, 2024

• Make local aqt support tiling the largest contraction axis (not the first) in the bwd.

  Show description for 12cc593
  

  ychzhang

  authored and

  copybara-github

  committed
  12cc593

  Copy full SHA for 12cc593

Commits on Jun 27, 2024

• Prevent GPTQ from accidentally being "activated" during TRAIN mode

  Show description for f224fdf
  

  cdh4696

  authored and

  copybara-github

  committed
  f224fdf

  Copy full SHA for f224fdf

Commits on Jun 26, 2024

• better SR

  Show description for 49cf7b8
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  49cf7b8

  Copy full SHA for 49cf7b8

• Version update for new AQT release.

  Show description for be46857
  

  cdh4696

  authored and

  copybara-github

  committed
  be46857

  Copy full SHA for be46857

Commits on Jun 25, 2024

• Add an optional Quantizer field for scale dtype.

  Show description for 8473116
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  8473116

  Copy full SHA for 8473116

• add fp numerics import in config

  Show description for afb7334
  

  ychzhang

  authored and

  copybara-github

  committed
  afb7334

  Copy full SHA for afb7334

• Optionally tile axes in tiled_dot_general

  Show description for e3d4a6c
  

  ychzhang

  authored and

  copybara-github

  committed
  e3d4a6c

  Copy full SHA for e3d4a6c

Commits on Jun 22, 2024

• General floating point numerics

  Show description for d6120c8
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  d6120c8

  Copy full SHA for d6120c8

Commits on Jun 21, 2024

• Separate a function to apply use_fwd_quant flag.

  Show description for 917e7a4
  

  cdh4696

  authored and

  copybara-github

  committed
  917e7a4

  Copy full SHA for 917e7a4

• Allow calibrate to return the updated lhs and rhs

  Show description for b12fbbf
  

  cdh4696

  authored and

  copybara-github

  committed
  b12fbbf

  Copy full SHA for b12fbbf

Commits on Jun 19, 2024

• Separate dimension number update function for backward

  Show description for b79050e
  

  cdh4696

  authored and

  copybara-github

  committed
  b79050e

  Copy full SHA for b79050e

• Refactor: Separate quant() function from dot_general_raw.

  Show description for ed49536
  

  cdh4696

  authored and

  copybara-github

  committed
  ed49536

  Copy full SHA for ed49536

Commits on Jun 18, 2024

• Move calibration instantiation in fwd pass to outside of custom_vjp to allow a stateful bwd pass calibrators.

  Show description for 330dd11
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  330dd11

  Copy full SHA for 330dd11

Commits on Jun 17, 2024

• Support auto sharding for subchannel on AQT level

  Show description for e8eec3a
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  e8eec3a

  Copy full SHA for e8eec3a

Commits on Jun 14, 2024

• New examples showcasing AQT usage for various quantization schemes.

  Show description for 60ecdf4
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  60ecdf4

  Copy full SHA for 60ecdf4

Commits on Jun 13, 2024

• Remove usage of xla_computation and replace with jax.jit(f).lower(*args).compile_ir()

  Show description for 5ff0bd4
  

  yashk2810

  authored and

  copybara-github

  committed
  5ff0bd4

  Copy full SHA for 5ff0bd4

Commits on Jun 7, 2024

• Pull out local_aqt from the aqt_dot_general"s __call__ function

  Show description for 81f708c
  

  cdh4696

  authored and

  copybara-github

  committed
  81f708c

  Copy full SHA for 81f708c

• Factor out tiling operations out of tiled_dot_general

  Show description for d8034eb
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  d8034eb

  Copy full SHA for d8034eb

Commits on Jun 5, 2024

• Make spatial_dimensions a required field and clarify in docstring

  Show description for 2afde45
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  2afde45

  Copy full SHA for 2afde45

Commits on Jun 4, 2024

• Fix FP8 target training loss

  Show description for f2cfe16
  

  ychzhang

  authored and

  copybara-github

  committed
  f2cfe16

  Copy full SHA for f2cfe16

Commits on Jun 3, 2024

• Added some TODOs

  Show description for 8068b53
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  8068b53

  Copy full SHA for 8068b53

Commits on May 31, 2024

• Make sure QTensor#dtype() returns an instance of jnp.dtype

  Show description for 702cb95
  

  Cerebra Catalyst Team

  authored and

  copybara-github

  committed
  702cb95

  Copy full SHA for 702cb95

Commits on May 30, 2024

• Update version to 0.7.4

  Show description for 4b3cf29
  

  cdh4696

  authored and

  copybara-github

  committed
  4b3cf29

  Copy full SHA for 4b3cf29

Pagination

Previous

Next