Code

Flash Attention for Neuron#939

Code

Open

apoorvtintin wants to merge 1 commit intoapple:mainapple/axlearn:mainfrom apoorvtintin:mainline_upstream_faapoorvtintin/axlearn:mainline_upstream_faCopy head branch name to clipboard

+338-2

Commits

Commits on Jan 23, 2025

• Flash Attention for Neuron

  apoorvtintin

  committed
  73a2808

  Copy full SHA for 73a2808