Comment by the__alchemist

I don't think that fits; that's a ML framework. The others in the link are general GPU frameworks.