MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI training and inference, such as FP8 row-wise quantization and collective communications.
☆152Sep 2, 2026Updated this week
Alternatives and similar repositories for MSLK
Users that are interested in MSLK are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Autonomous GPU kernel optimization system driven by AI agents.☆31Mar 29, 2026Updated 5 months ago
- Triton-based Symmetric Memory operators and examples☆110May 15, 2026Updated 3 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆367Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆931Updated this week
- Github mirror of trition-lang/triton repo.☆195Updated this week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- An Optimizer for Nvidia Compilers.☆130Aug 28, 2026Updated last week
- torchcomms: a modern PyTorch communications API☆391Updated this week
- ☆172Dec 27, 2024Updated last year
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Jul 17, 2026Updated last month
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 7 months ago
- A dynamic binary instrumentation tool for tracing and analyzing GPU kernel instructions.☆79Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆214Updated this week
- A Triton JIT runtime and ffi provider in C++☆40Aug 7, 2026Updated 3 weeks ago
- A Quirky Assortment of CuTe Kernels☆1,136Updated this week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- Low overhead tracing library and trace visualizer for pipelined CUDA kernels☆136Jul 29, 2026Updated last month
- NCU-driven iterative optimization workflow for CUDA/CUTLASS/Triton/CuTe DSL kernels.☆25Apr 10, 2026Updated 4 months ago
- ☆37Aug 7, 2025Updated last year
- Fast low-bit matmul kernels in Triton☆483Aug 9, 2026Updated 3 weeks ago
- a size profiler for cuda binary☆71Jan 15, 2026Updated 7 months ago
- incubator repo for CUDA-TileIR backend☆159Jul 10, 2026Updated last month
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆200Updated this week
- Kernels, of the mega variety :)☆817May 26, 2026Updated 3 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Perplexity GPU Kernels☆605Nov 7, 2025Updated 9 months ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆545Updated this week
- FLA but cuTile☆27Apr 17, 2026Updated 4 months ago
- Skills for writing tilelang and debugging with CUDA toolkits.☆137May 20, 2026Updated 3 months ago
- A pure-Python implementation of the Nvidia CuTe layout algebra intended to be approachable and easy to learn.☆241Jun 29, 2026Updated 2 months ago
- Helpful tools and examples for working with flex-attention☆1,234Updated this week
- Ship correct and fast LLM kernels to PyTorch☆154Jan 14, 2026Updated 7 months ago
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆580Aug 27, 2026Updated last week
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- TORCH_TRACE parser for PT2☆94May 11, 2026Updated 3 months ago
- Communication patterns for AI, built on top of NCCL device and host APIs☆43Updated this week
- Accelerating MoE with IO and Tile-aware Optimizations☆754Updated this week
- A place to store reusable transformer components of my own creation or found on the interwebs☆81Updated this week
- Distributed Compiler and Optimized Parallel Kernels☆1,532Aug 12, 2026Updated 3 weeks ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆200Feb 11, 2026Updated 6 months ago
- Open ABI and FFI for Machine Learning Systems☆455Updated this week