NVlabs / vibetensorLinks
Our first fully AI generated deep learning system
☆247Updated this week
Alternatives and similar repositories for vibetensor
Users that are interested in vibetensor are comparing it to the libraries listed below
Sorting:
- Autonomous GPU Kernel Generation via Deep Agents☆217Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆63Updated this week
- Ship correct and fast LLM kernels to PyTorch☆135Updated last week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆186Updated this week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆312Updated this week
- ☆117Updated 8 months ago
- ☆101Updated last year
- ring-attention experiments☆163Updated last year
- Helpful kernel tutorials and examples for tile-based GPU programming☆592Updated this week
- Accelerating MoE with IO and Tile-aware Optimizations☆553Updated last week
- Collection of kernels written in Triton language☆175Updated 9 months ago
- extensible collectives library in triton☆93Updated 9 months ago
- ☆273Updated this week
- TPU inference for vLLM, with unified JAX and PyTorch support.☆216Updated this week
- A bunch of kernels that might make stuff slower 😉☆75Updated this week
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆721Updated this week
- JAX backend for SGL☆227Updated this week
- Framework to reduce autotune overhead to zero for well known deployments.☆94Updated 4 months ago
- ☆133Updated 7 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆105Updated 6 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆140Updated 7 months ago
- DeeperGEMM: crazy optimized version☆73Updated 8 months ago
- Cataloging released Triton kernels.☆289Updated 4 months ago
- Fast low-bit matmul kernels in Triton☆423Updated last month
- mHC kernels implemented in CUDA☆233Updated last week
- Efficient Long-context Language Model Training by Core Attention Disaggregation☆80Updated 3 weeks ago
- The evaluation framework for training-free sparse attention in LLMs☆110Updated 3 months ago
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆440Updated last month
- ☆115Updated last year
- Distributed MoE in a Single Kernel [NeurIPS '25]☆188Updated this week