---
id: apache-tvm-ffi
version: "0.1.13.post3"
license: Apache 2.0
license_treatment: permissive
maintenance: active
---

# apache-tvm-ffi — tvm ffi

License: permissive · Maintenance: active · Downloads: 7.8M/mo

## What it is and what it does

Apache TVM FFI is an open ABI and FFI standard designed to let machine learning libraries, kernel DSLs, and runtimes interoperate across frameworks without recompilation. It defines a stable C ABI for exposing kernels and extensions, and provides Python bindings with zero-copy tensor interchange via the DLPack protocol. The package enables a single compiled wheel to work across PyTorch, JAX, CuPy, and NumPy, and supports multi-language bindings (Python, C++, Rust). It is currently in RFC status, meaning the core ABI is stable but the project is actively gathering community feedback before moving to semantic versioning.

The package depends only on typing-extensions and is available as pre-built wheels for modern Python versions (3.10–3.14) across Linux, macOS, and Windows. It is intended for kernel library authors, DSL framework developers, and ML infrastructure teams who need a framework-agnostic way to ship optimized code and integrate with multiple downstream consumers.

Use it for:

- Ship a single compiled kernel library wheel that works with PyTorch, JAX, and CuPy without rebuilding
- Expose a custom kernel DSL (like TileLang or cuteDSL) through a stable ABI for consumption by multiple frameworks
- Build ML infrastructure that accepts ABI-compliant extensions from third-party kernel libraries
- Enable zero-copy tensor passing between PyTorch and JAX using DLPack without data copying
- Integrate generated code from coding agents into production ML systems via a unified ABI

## Worth the install?

AI-flagged interpretation of the facts above — verify before relying.

Provides a stable, minimal C ABI and FFI for machine learning systems to expose kernels, DSLs, and runtime extensions across frameworks like PyTorch, JAX, and NumPy with zero-copy interop.

Yes, if you are building or consuming ML kernel libraries or DSLs that need to work across multiple frameworks. The stable C ABI, zero-copy interop, and multi-language support make it a solid choice for infrastructure teams. However, be aware that the project is in RFC stage and may evolve; if you require absolute API stability, wait for the first semantic-versioning release. No known security vulnerabilities.

## Install

pip install apache-tvm-ffi
uv add apache-tvm-ffi
poetry add apache-tvm-ffi

## Installing apache-tvm-ffi

Before you install: Medium install friction due to platform-specific wheels for multiple Python versions (3.10–3.14) and architectures (x86_64, aarch64, arm64, Windows). Active maintenance with a release 4 days old and recent commits; early-stage RFC status means the ABI is stable but the project is still soliciting community feedback.

License in practice: Apache 2.0 permissive license allows use in most commercial and open-source projects without significant restriction, though derivative works must include license notices.

Quickstart:

pip install apache-tvm-ffi

import tvm.ffi
# Use TVM FFI to expose or consume ABI-compliant kernels across frameworks

Requires Python >= 3.9; pre-built wheels available for Python 3.10–3.14 on Linux (manylinux), macOS (arm64), and Windows; other configurations may require building from source.

Verify before relying:

- Whether the RFC stage will introduce breaking C ABI changes before semantic versioning begins
- Performance overhead of the FFI layer compared to direct framework-specific APIs
- Completeness of multi-language support (C++, Rust) in the current release

## Package facts

- License: Apache 2.0 (permissive)
- Python support: supports_current
- Install friction: medium
- Maintenance: active
- Downloads: 7.8M/month (top 5,000 on PyPI)
- Known vulnerabilities: none known

## Tags

machine learning ABI FFI, kernel library interop, pytorch jax numpy bridge, zero-copy tensor interchange, ml framework extension, dlpack interoperability, kernel dsl runtime, ml-interop, kernel-library, ffi-abi

[View on SkillFed](https://skillfed.io/packages/apache-tvm-ffi) · [View on PyPI](https://pypi.org/project/apache-tvm-ffi/)