sshan-zhao / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Home Page:http://www.tvmlang.org/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Open Deep Learning Compiler Stack

GitHub license Build Status

Installation | Documentation | Tutorials | Operator Inventory | FAQ | Contributors | Community | Release Notes

TVM is a compiler stack for deep learning systems. It is designed to close the gap between the productivity-focused deep learning frameworks, and the performance- and efficiency-focused hardware backends. TVM works with deep learning frameworks to provide end to end compilation to different backends. Checkout the tvm stack homepage for more information.

License

© Contributors Licensed under an Apache-2.0 license.

Contribute to TVM

TVM adopts apache committer model, we aim to create an open source project that is maintained and owned by the community.

Acknowledgement

We learnt a lot from the following projects when building TVM.

  • Halide: TVM uses HalideIR as data structure for arithematic simplification and low level lowering. We also learnt and adapted some part of lowering pipeline from Halide.
  • Loopy: use of integer set analysis and its loop transformation primitives.
  • Theano: the design inspiration of symbolic scan operator for recurrence.

About

Open deep learning compiler stack for cpu, gpu and specialized accelerators

http://www.tvmlang.org/

License:Apache License 2.0


Languages

Language:C++ 55.6%Language:Python 34.3%Language:Java 4.0%Language:JavaScript 1.3%Language:Objective-C++ 1.2%Language:C 0.8%Language:Makefile 0.7%Language:Verilog 0.7%Language:Shell 0.6%Language:CMake 0.3%Language:Objective-C 0.3%Language:HTML 0.1%