triton-msl 0.1.0a2 — portability verified on NVIDIA

First public alpha with the portability story verified on real NVIDIA silicon.

pip install triton-msl==0.1.0a2   # import triton_msl
pip install "triton>=3.6.0"       # Triton, installed separately (build from source on macOS)

Highlights

• Portability — verified. The same unmodified @triton.jit kernels run on Apple Metal and NVIDIA (checked on a rented A40): vector-add and the fp32/ieee matmul were bit-identical across vendors; softmax matched to fp rounding (~1e-9). Develop and correctness-debug on your Mac; it's the same code that runs on CUDA. See PORTABILITY.md.
• Correctness — the anti-silent-wrong campaign. ~75 silent-wrong bugs closed across the dot/reduce/store surfaces (plus three fuzzers); the dot and reduce paths are now correct-or-refuse by construction — general address-traced matmul stride inference, a structural reduce-combine classifier, masked-store correctness.
• torch.compile — a persistent MSL stash makes warm GPT-2 small ~2.1 ms (~24×), faster than PyTorch's own native-MPS Inductor path; NaN-propagating reductions, CNN/BatchNorm, fused reduce+scan.
• Performance (M4 Max) — matmul ~11–12 TFLOP/s (fp32/fp16/bf16), FlashAttention head_dim 128 at 5.1/6.3 TFLOP/s, zero-copy memory-bound ~315–347 GB/s.
• Integrity contract — kernels we can lower run correctly; those we can't are refused loudly (MetalNonRecoverableError), never silent-wrong.

Project suite 1,968 / 0; upstream Triton test_core conformance 5,560 / 0 / 3,782 skips.

Full details: CHANGELOG.md.

Triton 3.7.0 prebuilt wheel — macOS arm64 (cp314)

Prebuilt Triton wheel — macOS / Apple Silicon (unofficial convenience build)

Triton publishes no macOS wheel, so using triton-msl normally means building Triton from source once (pip install git+https://github.com/triton-lang/triton.git, ~12 min). This is a prebuilt wheel so you can skip that step.

Install

pip install https://github.com/bledden/triton-msl/releases/download/triton-wheel-3.7.0-cp314-macos-arm64/triton-3.7.0+git4da2e268-cp314-cp314-macosx_15_0_arm64.whl

Scope — this exact wheel fits

• Python 3.14 (cp314)
• macOS 15.0 or newer
• Apple Silicon / arm64 — one wheel covers M1 through M5; the Triton frontend is chip-agnostic, and triton-msl adapts to your GPU at runtime.

If any of those differ (other Python version, older macOS, Intel Mac), build from source — that path stays the primary, supported one.

Provenance

• Built from triton-lang/triton @ 4da2e268 (Triton 3.7.0).
• LLVM is statically linked into libtriton.so, so the wheel is self-contained — verified to import in a clean, source-free venv.
• The wheel bundles Triton's NVIDIA backend, which is inert on macOS; triton-msl supplies its own Metal backend.

Unofficial. Not published by triton-lang. A convenience artifact for triton-msl users on this exact platform tuple — build-from-source remains the primary path.