Code

Replace exp2 with exp#25984

Code

Merged

copybara-service[bot] merged 2 commits intojax-ml:mainjax-ml/jax:mainfrom Rifur13:use_exp2Rifur13/jax:use_exp2Copy head branch name to clipboardJan 20, 2025

+26-14

Commits

Commits on Jan 19, 2025

• Speed up attention kernel by using exp2

  Rifur13

  committed
  1f59506

  Copy full SHA for 1f59506

Commits on Jan 20, 2025

• Merge branch 'jax-ml:main' into use_exp2

  Rifur13

  authored
  e7db4d5

  Copy full SHA for e7db4d5