TVM - это открытый стек компилятора глубокого обучения для процессоров, графических процессоров и специализированных ускорителей. Он направлен на сокращение разрыва между ориентированными на продуктивность структурами глубокого обучения и аппаратными бэкэндами, ориентированными на производительность или эффективность. TVM предоставляет следующие основные функции:

  • Компиляция моделей глубокого обучения в Keras, MXNet, PyTorch, Tensorflow, CoreML, DarkNet в минимально развертываемые модули на различных аппаратных серверах.
  • Инфраструктура для автоматического создания и оптимизации тензорных операторов на большем количестве серверных модулей с более высокой производительностью.

TVM обеспечивает двухуровневую оптимизацию, показанную на следующем рисунке. Оптимизация вычислительного графа для выполнения таких задач, как объединение операторов высокого уровня, преобразование макета и управление памятью. Затем слой оптимизации тензорных операторов и генерации кода, который оптимизирует тензорные операторы.

TVM был принят в Apache Incubator, и я буду наставником проекта Apache TVM. Я буду наблюдать за сообществом TVM, чтобы они смогли достичь открытой меритократической среды.

[1] https://tvm.ai

[2] https://medium.com/@furkankamaci/open-source-software-development-and-apache-incubator-372cc90081ae