Code

Utils for fp8 gated einsum #4221

Code

Open

wenscarl wants to merge 7 commits intogoogle:maingoogle/flax:mainfrom wenscarl:gated_direct_quant_flaxwenscarl/flax:gated_direct_quant_flaxCopy head branch name to clipboard

+305-205

Commits

Commits on Sep 23, 2024

• design 1

  wenscarl

  committed
  a4d0265

  Copy full SHA for a4d0265

• oneside

  wenscarl

  committed
  60a9a91

  Copy full SHA for 60a9a91

• pass test

  wenscarl

  committed
  920099e

  Copy full SHA for 920099e

• pass test

  wenscarl

  committed
  c9b0fc5

  Copy full SHA for c9b0fc5

Commits on Sep 24, 2024

• fix quantize with scale type

  wenscarl

  committed
  9d0f844

  Copy full SHA for 9d0f844

• fix scale dtype and add decorator

  wenscarl

  committed
  de8d158

  Copy full SHA for de8d158

• all in decorator

  wenscarl

  committed
  2bf7694

  Copy full SHA for 2bf7694