Alternatives and similar repositories for nki-llama

Users that are interested in nki-llama are comparing it to the libraries listed below

• aws-neuron / nki-samples
  ☆47Updated this week
• aws-neuron / neuronx-distributed
  ☆60Updated last week
• awslabs / nki-autotune
  ☆15Updated this week
• awslabs / slapo
  A schedule language for large model training
  ☆151Updated last month
• yifuwang / symm-mem-recipes
  ☆121Updated 9 months ago
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆223Updated last week
• parasailteam / coconet
  ☆83Updated 2 years ago
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆581Updated 2 weeks ago
• mk1-project / quickreduce
  QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.
  ☆33Updated last month
• Deep-Learning-Profiling-Tools / triton-viz
  ☆242Updated this week
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆78Updated this week
• ParCIS / Chimera
  Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.
  ☆66Updated 6 months ago
• ColfaxResearch / cutlass-kernels
  ☆238Updated last year
• AMD-AGI / Primus-Turbo
  ☆22Updated last week
• perplexityai / pplx-kernels
  Perplexity GPU Kernels
  ☆476Updated 2 weeks ago
• ColfaxResearch / cfx-article-src
  ☆144Updated 4 months ago
• mlc-ai / mlc-python
  ☆37Updated 2 months ago
• UofT-EcoSystem / DietCode
  DietCode Code Release
  ☆65Updated 3 years ago
• AlibabaResearch / mononn
  ☆30Updated last year
• gpu-mode / triton-index
  Cataloging released Triton kernels.
  ☆261Updated 3 weeks ago
• openxla / shardy
  MLIR-based partitioning system
  ☆135Updated this week
• osayamenja / FlashMoE
  Distributed MoE in a Single Kernel [NeurIPS '25]
  ☆49Updated this week
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆221Updated 2 years ago
• sunlex0717 / DissectingTensorCores
  ☆108Updated last year
• Raphael-Hao / brainstorm
  Compiler for Dynamic Neural Networks
  ☆46Updated last year
• ParCIS / Magicube
  Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.
  ☆89Updated 2 years ago
• ColfaxResearch / layout-categories
  This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".
  ☆48Updated last week
• awslabs / ratex
  ☆23Updated last month
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆328Updated this week
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 4 months ago