Apache TVM vs NVIDIA cuDNN

Apache TVM is a machine learning compilation framework that turns pre-trained models into deployable modules. Its optimization process can be customized in Python without rebuilding the TVM stack, and users can combine optimization passes, libraries, and code generation. It imports models from PyTorch, ONNX, and TensorFlow Lite. Target backends include CPU, GPU, Metal, ROCm, Vulkan, OpenCL, x86, ARM, and WebAssembly. A lightweight runtime can execute compiled code in JavaScript, Java, Python, and C++ on Android, iOS, Raspberry Pi, and web browsers. Users can install it from PyPI, build from source, or use Docker images; cross-compilation and RPC deployment cover ARM, x86, RISC-V, embedded systems, and accelerator devices. The default generated binary uses a minimal runtime API and limited system calls such as malloc. Apache TVM is free open-source software under Apache License 2.0. Its RPC server assumes trusted users and networks: API users can write arbitrary files and obtain full remote code execution, so it should be treated as a high-trust interface.

NVIDIA cuDNN has no write-up on Specifiction yet; its figures are above.