Code

Flash Attention for Neuron#883

Code

Open

apoorvtintin wants to merge 1 commit intoapple:mainapple/axlearn:mainfrom ptoulme-aws:mainline_upstream_faptoulme-aws/axlearn:mainline_upstream_faCopy head branch name to clipboard

+272-1

Commits

Commits on Dec 11, 2024

• Flash Attention for Neuron

  apoorvtintin

  committed
  347f522