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.
☆155Oct 6, 2026Updated last 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.☆32Mar 29, 2026Updated 6 months ago
- Triton-based Symmetric Memory operators and examples☆113May 15, 2026Updated 4 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆372Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆963Updated this week
- Github mirror of trition-lang/triton repo.☆201Updated this week
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- An Optimizer for Nvidia Compilers.☆137Sep 23, 2026Updated 3 weeks ago
- torchcomms: a modern PyTorch communications API☆401Updated this week
- ☆171Dec 27, 2024Updated last year
- FA4-based Relative Attention Kernel developed by TML and Colfax☆18Sep 11, 2026Updated last month
- High performance RMSNorm Implement by using SM Core Storage(Registers and Shared Memory)☆31Jan 22, 2026Updated 8 months ago
- A dynamic binary instrumentation tool for tracing and analyzing GPU kernel instructions.☆83Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆229Updated this week
- A Triton JIT runtime and ffi provider in C++☆41Sep 24, 2026Updated 2 weeks ago
- A Quirky Assortment of CuTe Kernels☆1,173Sep 17, 2026Updated 3 weeks ago
- 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
- Unified tracing profiler and visualizer, one timeline from CPU to GPU/HES to in-kernel zones☆148Sep 21, 2026Updated 3 weeks ago
- NCU-driven iterative optimization workflow for CUDA/CUTLASS/Triton/CuTe DSL kernels.☆25Apr 10, 2026Updated 6 months ago
- ☆37Aug 7, 2025Updated last year
- Fast low-bit matmul kernels in Triton☆488Oct 1, 2026Updated last week
- a size profiler for cuda binary☆73Oct 3, 2026Updated last week
- incubator repo for CUDA-TileIR backend☆164Updated this week
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆203Sep 2, 2026Updated last month
- Kernels, of the mega variety :)☆847May 26, 2026Updated 4 months ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Perplexity GPU Kernels☆608Nov 7, 2025Updated 11 months ago
- Benchmark tests supporting the TiledCUDA library.☆19Nov 19, 2024Updated last year
- Helpful tools and examples for working with flex-attention☆1,256Updated this week
- FLA but cuTile☆27Apr 17, 2026Updated 5 months ago
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆555Updated this week
- Skills for writing tilelang and debugging with CUDA toolkits.☆145Sep 15, 2026Updated 3 weeks ago
- A pure-Python implementation of the Nvidia CuTe layout algebra intended to be approachable and easy to learn.☆253Jun 29, 2026Updated 3 months ago
- Kernel Library for Large Headdim Attention (64~1024, BF16/FP8/FP4), 1.5x~15x↑ vs PyTorch SDPA.☆340Updated this week
- Ship correct and fast LLM kernels to PyTorch☆155Jan 14, 2026Updated 9 months ago
- 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.
- TORCH_TRACE parser for PT2☆94May 11, 2026Updated 5 months ago
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆597Oct 3, 2026Updated last week
- Accelerating MoE with IO and Tile-aware Optimizations☆779Aug 29, 2026Updated last month
- A place to store reusable transformer components of my own creation or found on the interwebs☆81Oct 4, 2026Updated last week
- Distributed Compiler and Optimized Parallel Kernels☆1,566Sep 18, 2026Updated 3 weeks ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆206Feb 11, 2026Updated 8 months ago
- Open ABI and FFI for Machine Learning Systems☆470Updated this week