Kernels Python Packages

spyder-kernels

Jupyter Kernels for the Spyder console

ttnn

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

greyhound-kernels

CuTe DSL kernels for PyTorch training and inference workloads

tensordev

JAX-based tensor algebra for signatures, Volterra signatures, free developments, and signature kernels.

gpuemu

Catch silently-wrong GPU kernels before they reach production.

kernelmethods

Kernel methods and classes for composable kernel learning workflows.

rfx-sdk

A tinygrad-inspired robotics framework for Unitree Go2

rustkernels

Python bindings for RustKernels GPU-accelerated kernel library

rkhs

A unified framework for marginal and conditional two-sample testing with kernels in JAX.

graphkke

Library for the analysis of graph-structured and tree-structured data