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
  ☆51Updated 3 weeks ago
• aws-neuron / neuronx-distributed
  ☆60Updated last month
• awslabs / nki-autotune
  ☆15Updated this week
• yifuwang / symm-mem-recipes
  ☆141Updated 9 months ago
• perplexityai / pplx-kernels
  Perplexity GPU Kernels
  ☆497Updated last month
• awslabs / slapo
  A schedule language for large model training
  ☆151Updated 2 months ago
• aws-samples / sagemaker-trainium-inferentia
  ☆14Updated last year
• ColfaxResearch / cutlass-kernels
  ☆241Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆215Updated this week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆612Updated 2 weeks ago
• mk1-project / quickreduce
  QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.
  ☆33Updated last month
• parasailteam / coconet
  ☆83Updated 2 years ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆242Updated this week
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆86Updated last week
• AMD-AGI / Primus-Turbo
  ☆27Updated this week
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆491Updated this week
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆226Updated last week
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆299Updated 2 months ago
• ColfaxResearch / cfx-article-src
  ☆150Updated 5 months ago
• sitar-lab / NeuSight
  ☆53Updated 4 months ago
• yanring / Megatron-MoE-ModelZoo
  Best practices for training DeepSeek, Mixtral, Qwen and other MoE models using Megatron Core.
  ☆114Updated last week
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆377Updated last month
• Shenggan / awesome-distributed-ml
  A curated list of awesome projects and papers for distributed training or inference
  ☆247Updated last year
• AlibabaResearch / flash-llm
  Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity
  ☆221Updated 2 years ago
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 5 months ago
• triton-lang / kernels
  ☆92Updated 11 months ago
• zhuohan123 / terapipe
  ☆75Updated 4 years ago
• microsoft / nnscaler
  nnScaler: Compiling DNN models for Parallel Training
  ☆117Updated last month
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆41Updated 2 years ago
• awslabs / ratex
  ☆23Updated 2 months ago