https://github.com/apache/tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://github.com/apache/tvm

compiler deep-learning gpu javascript machine-learning metal opencl performance rocm spirv tensor tvm vulkan

Last synced: 5 days ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/apache/tvm
• Owner: apache
• License: apache-2.0
• Created: 2016-10-12T22:20:28.000Z (about 8 years ago)
• Default Branch: main
• Last Pushed: 2024-11-22T05:03:09.000Z (29 days ago)
• Last Synced: 2024-11-22T09:07:12.781Z (29 days ago)
• Topics: compiler, deep-learning, gpu, javascript, machine-learning, metal, opencl, performance, rocm, spirv, tensor, tvm, vulkan
• Language: Python
• Homepage: https://tvm.apache.org/
• Size: 102 MB
• Stars: 11,809
• Watchers: 376
• Forks: 3,474
• Open Issues: 684
• Metadata Files:
  • Readme: README.md
  • Changelog: NEWS.md
  • License: LICENSE
  • Codeowners: .github/CODEOWNERSHIP

Awesome Lists containing this project

• awesome-deep-model-compression - TVM (Apache) - glr-tutorials-frontend-from-tensorflow-py)] [](https://github.com/apache/tvm) (Tools / Hard-ware Integration)
• awesome-list - Apache TVM - Open deep learning compiler stack for cpu, gpu and specialized accelerators. (Deep Learning Framework / Deployment & Distribution)
• awesome-yolo-object-detection - TVM
• awesome-yolo-object-detection - TVM
• awesome-llmops - TVM - square) | (Performance / ML Compiler)

README

        


 Open Deep Learning Compiler Stack

==============================================

[Documentation](https://tvm.apache.org/docs) |

[Contributors](CONTRIBUTORS.md) |

[Community](https://tvm.apache.org/community) |

[Release Notes](NEWS.md)

[![Build Status](https://ci.tlcpack.ai/buildStatus/icon?job=tvm/main)](https://ci.tlcpack.ai/job/tvm/job/main/)

[![WinMacBuild](https://github.com/apache/tvm/workflows/WinMacBuild/badge.svg)](https://github.com/apache/tvm/actions?query=workflow%3AWinMacBuild)

Apache 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.

License

-------

TVM is licensed under the [Apache-2.0](LICENSE) license.

Getting Started

---------------

Check out the [TVM Documentation](https://tvm.apache.org/docs/) site for installation instructions, tutorials, examples, and more.

The [Getting Started with TVM](https://tvm.apache.org/docs/tutorial/introduction.html) tutorial is a great

place to start.

Contribute to TVM

-----------------

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

Check out the [Contributor Guide](https://tvm.apache.org/docs/contribute/).

Acknowledgement

---------------

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

- [Halide](https://github.com/halide/Halide): Part of TVM's TIR and arithmetic simplification module

  originates from Halide. We also learned and adapted some part of lowering pipeline from Halide.

- [Loopy](https://github.com/inducer/loopy): use of integer set analysis and its loop transformation primitives.

- [Theano](https://github.com/Theano/Theano): the design inspiration of symbolic scan operator for recurrence.