Understanding DeepMind and Strassen algorithms

An introduction to the matrix multiplication problem, with applications in Python and JAX

Image by Ivan Diaz on Unsplash

DeepMind recently published an interesting paper that employed Deep Reinforcement Learning to find new Matrix Multiplication algorithms[1]. One of the objectives of this paper is to lower matrix multiplication computational complexity…