kurtosis/mat_mul

Implementation of "Discovering faster matrix multiplication algorithms with reinforcement learning" paper from DeepMind

mat_mul (AlphaTensor)

Implementation of AlphaTensor ("Discovering faster matrix multiplication algorithms with reinforcement learning") from DeepMind. See blog post.

Getting started

Run training.py to train an AlphaTensor model.