The code has been tested running under Python 3.7.4, with the following packages and their dependencies installed: numpy==1.16.5 matplotlib==3.1.0 sklearn==0.21.3 seaborn==0.11.2 pytorch==1.7.1 ...
If you see the official example, it is really confusing to use where to start. Use tritony! You will get really short lines of code like example below.