d4l3k / axe

A simple graph partitioning algorithm written in Go. Designed for use for partitioning neural networks across multiple devices which has an added cost when crossing device boundaries.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

d4l3k/axe Stargazers