Menu
Apache TVM
Apache TVM is an open-source ML compiler framework for FPGAs, microcontrollers, browsers, CPUs, and GPUs. It works with popular frameworks like PyTorch and TensorFlow. The product was developed by OctoML to help engineers run computations efficiently on any hardware backend.ย
Project Background
- Framework:ย Apache TVM
- Author: OctoML
- Released: 2017
- Type: Open-source ML compiler framework for CPUs, GPUs, and acceleratorsย
- License: Apache 2.0
- Supports: Works with PyTorch, TensorFlow, Keras, MXNet, and others
- Language: C++ and Python
- GitHub:ย apache/tvmย with 7k contributors and 591 contributors
Features
- A universal compiler that ingests data from different frameworks and enables them to work on different devices from CPUs to GPUs to mobile devices to browsersย
- End to end deep learning compilation stack
- Problem it addresses: 1) Portability-allows users to deploy on different devicesย 2) Efficiency – squeezes as much performance as possible from hardware 3) Support-built a stack that works with hardwareย
ย
