Alternatives and similar repositories for nki-autotune

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

• aws-neuron / nki-samples
  ☆47Updated last week
• aws-neuron / neuronx-distributed
  ☆60Updated this week
• aws-neuron / nki-llama
  Project showing how to develop NKI kernels for Llama 3.2 1B inference
  ☆19Updated 3 months ago
• aws-neuron / aws-neuron-sdk
  Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and i…
  ☆537Updated last week
• aws-neuron / neuronx-nemo-megatron
  ☆39Updated 8 months ago
• aws-neuron / neuronx-distributed-inference
  ☆19Updated last 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 …
  ☆214Updated this week
• aws-neuron / transformers-neuronx
  ☆111Updated 7 months ago
• aws-neuron / aws-neuron-samples
  Example code for AWS Neuron SDK developers building inference and training applications
  ☆149Updated last week
• albanD / subclass_zoo
  ☆174Updated last year
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆546Updated last week
• rwitten / HighPerfLLMs2024
  ☆530Updated last year
• aws-samples / awsome-distributed-training
  Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.
  ☆335Updated this week
• awslabs / slapo
  A schedule language for large model training
  ☆149Updated 2 weeks ago
• huggingface / optimum-neuron
  Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.
  ☆239Updated this week
• aws-samples / aws-efa-nccl-baseami-pipeline
  EFA/NCCL base AMI build Packer and CodeBuild/Pipeline files. Also base Docker build files to enable EFA/NCCL in containers
  ☆44Updated last year
• openxla / stablehlo
  Backward compatible ML compute opset inspired by HLO/MHLO
  ☆531Updated 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.
  ☆209Updated 2 weeks ago
• aws-neuron / upstreaming-to-vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆19Updated last month
• aws-samples / awsome-inference
  ☆52Updated this week
• HazyResearch / aisys-building-blocks
  Building blocks for foundation models.
  ☆541Updated last year
• aws-samples / ec2-topology-aware-for-slurm
  ☆13Updated 3 months ago
• perplexityai / pplx-kernels
  Perplexity GPU Kernels
  ☆451Updated last month
• yifuwang / symm-mem-recipes
  ☆115Updated 8 months ago
• cli99 / llm-analysis
  Latency and Memory Analysis of Transformer Models for Training and Inference
  ☆450Updated 4 months ago
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆293Updated 2 weeks ago
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆406Updated 2 weeks ago
• mobiusml / gemlite
  Fast low-bit matmul kernels in Triton
  ☆356Updated last week
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆280Updated this week
• aws-neuron / neuronx-distributed-training
  ☆12Updated 3 weeks ago