search

google-research / dex-lang

Research language for array processing in the Haskell/ML family
BSD 3-Clause "New" or "Revised" License
1.56k stars 106 forks source link

Actually code the custom linearization for matmul #1295

Closed axch closed 1 year ago

axch commented 1 year ago

so that we don't embarrass ourselves too much with how poorly AD does thereon.