uTLX — Triton Language Extensions (Plugin)

utlx ships a Triton plugin (libutlx.so) plus the utlx Python DSL that exposes the TLX dialect (async loads, warp-group MMA, shared/tensor-memory buffers, etc.).

Compatibility

This wheel contains a native libutlx.so that links against libtriton.so and is ABI-coupled to a specific Triton build. Build Triton from the upstream triton-lang/triton repository at the v3.7.0 tag:

git clone https://github.com/triton-lang/triton
cd triton && git checkout v3.7.0
pip install -e . --no-build-isolation

Note: at the v3.7.0 tag the plugin custom-op binding in python/src/ir.cc discards the op result, so TLX ops return None. Apply the one-line fix (prepend an empty result slot, call addOp, return args[0]) so ops return their created value.

Requirements

• Linux x86_64, glibc >= 2.14, libstdc++ from GCC 11+ (GLIBCXX_3.4.29).
• Set TRITON_PLUGIN_PATHS to the installed libutlx.so and build Triton with TRITON_EXT_ENABLED=ON.

License

MIT. Copyright (c) 2026 Meta Platforms, Inc. and affiliates.