kurtosis / mat_mul Public

Notifications You must be signed in to change notification settings
Fork 1
Star 5

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

5 stars 1 fork Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 36 Commits
notebooks		notebooks
.gitignore		.gitignore
README.md		README.md
__init__.py		__init__.py
act.py		act.py
datasets.py		datasets.py
model.py		model.py
requirements.txt		requirements.txt
tests.py		tests.py
training.py		training.py
utils.py		utils.py

Repository files navigation

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.

About

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

Report repository

Releases

No releases published

Packages

No packages published

Languages