Implementation of Strassen attention, from Kozachinskiy et al. of National Center of AI in Chile - View it on GitHub