A pure-Python implementation of the Nvidia CuTe layout algebra intended to be approachable and easy to learn.
☆240Jun 29, 2026Updated last month
Alternatives and similar repositories for tensor-layouts
Users that are interested in tensor-layouts are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Interactive version of the CuTe layout paper☆57Apr 14, 2026Updated 4 months ago
- A Quirky Assortment of CuTe Kernels☆1,117Updated this week
- Region-level profiling for CUDA kernels with trace, NVBit, CUPTI, NSys, and an interactive Explorer.☆127Apr 17, 2026Updated 4 months ago
- Low overhead tracing library and trace visualizer for pipelined CUDA kernels☆135Jul 29, 2026Updated 2 weeks ago
- A Python DSL to write Nvidia PTX for Hopper and Blackwell in JAX and PyTorch☆374Jul 9, 2026Updated last month
- 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.
- This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".☆142Updated this week
- Cute layout visualization☆45Jan 18, 2026Updated 7 months ago
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆213Updated this week
- An experimental communicating attention kernel based on DeepEP.☆34Jul 29, 2025Updated last year
- DeeperGEMM: crazy optimized version☆86May 5, 2025Updated last year
- High-performance GPU kernels for Ads and Recsys model training, independently implemented and optimized for real-world workloads and mode…☆37Aug 7, 2026Updated last week
- Open ABI and FFI for Machine Learning Systems☆446Updated this week
- Github mirror of trition-lang/triton repo.☆188Updated this week
- Accelerating MoE with IO and Tile-aware Optimizations☆745Updated 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.
- ☆129May 10, 2026Updated 3 months ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆922Updated this week
- ☆32Jul 2, 2025Updated last year
- From Minimal GEMM to Everything☆234Jul 9, 2026Updated last month
- ☆40Dec 14, 2025Updated 8 months ago
- Skills for writing tilelang and debugging with CUDA toolkits.☆133May 20, 2026Updated 2 months ago
- CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-base…☆1,008Jul 22, 2026Updated 3 weeks ago
- ☆91Feb 5, 2026Updated 6 months ago
- ☆383Updated 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.
- FLA but cuTile☆27Apr 17, 2026Updated 4 months ago
- Communication patterns for AI, built on top of NCCL device and host APIs☆28Updated this week
- tutorials about polyhedral compilation.☆71Jun 6, 2026Updated 2 months ago
- ☆198Aug 9, 2026Updated last week
- PTX ISA 9.1 documentation converted to searchable markdown. Includes Claude Code skill for CUDA development.☆224Dec 24, 2025Updated 7 months ago
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆575Aug 10, 2026Updated last week
- mKernel: fast multi-node, multi-GPU fused kernels☆266Updated this week
- ☆65Apr 26, 2025Updated last year
- Perplexity GPU Kernels☆601Nov 7, 2025Updated 9 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools☆286Updated this week
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆198Feb 11, 2026Updated 6 months ago
- Fastest kernels written from scratch☆606Updated this week
- An Optimizer for Nvidia Compilers.☆128Updated this week
- FlashInfer Bench @ MLSys 2026: Building AI agents to write high performance GPU kernels☆181Apr 26, 2026Updated 3 months ago
- ☆51Updated this week
- CUDA kernels for linear attention variants, written in CuTe DSL and CUTLASS C++.☆538Updated this week