Note that TVM (<a href="https://github.com/dmlc/tvm" rel="nofollow">https://github.com/dmlc/tvm</a>) is a similar system (the IR is a fork of the Halide IR, but substantially improved), and is much more suited to deep learning, with tools like:<p><pre><code> - TOPI, a library of optimized routines for common deep learning operations
- Importers for TensorFlow, ONNX (PyTorch/Caffe2), Keras, MXNet, DarkNet and other frameworks.
- AutoTVM for automatically finding fast schedules for arbitrary devices, much faster than random search or the automatic schedules generated by e.g. Halide.
- Various mobile GPU runtimes (OpenGL, OpenCL, Vulkan, etc), compared to Halide
- Large community contributing optimized runtimes (e.g. Intel + Amazon contributing CPU improvements, improved schedules/declarations, etc)
</code></pre>
Highly recommend checking it out.