PyRank
  • Search
  • PyPI
  • GitHub
  • Insights
  • Compare
  • Advisories
  • Ecosystem
  • About

Kernel Fusion Python Packages

Python packages with the GitHub topic kernel-fusion. Sorted by monthly downloads, with stars and forks.
ROCm
tokenspeed-iris

AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming

3K 202 52
nactttch
g2n

A torch.compile backend that pays attention: custom FX fusion passes, a Triton LayerNorm kernel, and a persistent compile cache. Apache-2.0 free core of the g2n platform.

1K 0 0
svdrecbd
mhc-mlx

MLX + Metal implementation of mHC: Manifold-Constrained Hyper-Connections by DeepSeek-AI.

341 3 1
falcons-eyes
cutile-gpt

A complete GPT built entirely on NVIDIA cuTile: declarative GPU kernels in ~20 lines instead of ~150 hand-tuned CUDA, matching PyTorch speed in a ~10MB footprint

317 7 1
TxsharDev
alia-molten

Write the math. Get the kernel. Fused CUDA kernel generation from mathematical specifications.

172 1 0
Argonaut790
fused-turboquant

Fused Triton kernels for TurboQuant KV cache compression — 2-4 bit quantization with RHT rotation. Drop-in HuggingFace & vLLM integration. Up to 4.9x KV cache compression for Llama, Qwen, Mistral, and more.

74 8 1
    • Original data from PyPI, GitHub, and OSV; downloads served through ClickHouse
    • Sources & methodology