Hidet is an open-source deep learning compiler, written in Python. It supports end-to-end compilation of DNN models from PyTorch and ONNX to efficient cuda kernels. A series of graph-level and ...
It is tested on Python 2.7, 3.4, 3.5 and 3.6. It requires numpy and scipy.