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
  ☆29Updated this week
• awslabs / nki-autotune
  ☆11Updated last week
• aws-neuron / neuronx-distributed
  ☆52Updated last month
• awslabs / slapo
  A schedule language for large model training
  ☆145Updated 9 months ago
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 3 months ago
• openxla / shardy
  MLIR-based partitioning system
  ☆74Updated this week
• triton-lang / kernels
  ☆73Updated 4 months ago
• awslabs / ratex
  ☆23Updated 4 months ago
• awslabs / lorien
  ☆44Updated last year
• parasailteam / coconet
  ☆76Updated 2 years ago
• cchan / tccl
  extensible collectives library in triton
  ☆84Updated 6 months ago
• zhisbug / Cavs
  Cavs: An Efficient Runtime System for Dynamic Neural Networks
  ☆14Updated 4 years ago
• asplos-contest / 2025
  The ASPLOS 2025 / EuroSys 2025 Contest Track
  ☆30Updated last week
• tbd-ai / tbd-suite
  ☆47Updated 2 years ago
• mlc-ai / docs
  The documents for TVM Unity
  ☆9Updated 7 months ago
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆25Updated this week
• spcl / mlir-dace
  Data-Centric MLIR dialect
  ☆40Updated last year
• aws-neuron / neuronx-nemo-megatron
  ☆35Updated 3 months ago
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆237Updated last week
• NVlabs / nvbitfi
  Architecture-level Fault Injection Tool for GPU Application Resilience Evaluation
  ☆58Updated last year
• yifuwang / symm-mem-recipes
  ☆57Updated 3 months ago
• Azure / msccl
  Microsoft Collective Communication Library
  ☆60Updated 4 months ago
• uchuhimo / amanda
  ☆18Updated 11 months ago
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆267Updated this week
• albanD / subclass_zoo
  ☆162Updated 9 months ago
• ParCIS / Chimera
  Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.
  ☆61Updated last week
• ColfaxResearch / cfx-article-src
  ☆90Updated 2 weeks ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆189Updated last week
• dglai / FeatGraph
  Sparse kernels for GNNs based on TVM
  ☆16Updated 4 years ago
• intel / torch-xpu-ops
  ☆37Updated this week