Inspired by the outstanding paper "AlphaTensor: Discovering faster matrix multiplication algorithms with reinforcement learning", in this repo I used pyomo library and IPOPT optimizer to solve the problem of finding a correct matrix multiplication algorithm with a predifined number of scalar multiplications.
sajadalipour7/Discovering-faster-matrix-multiplication-algorithm-with-optimization
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|