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-llama
  Project showing how to develop NKI kernels for Llama 3.2 1B inference
  ☆20Updated 8 months ago
• aws-neuron / nki-samples
  ☆58Updated last week
• aws-neuron / neuronx-distributed
  ☆64Updated last month
• 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…
  ☆573Updated last week
• aws-neuron / neuronx-nemo-megatron
  ☆39Updated last year
• aws-neuron / aws-neuron-samples
  Example code for AWS Neuron SDK developers building inference and training applications
  ☆156Updated 2 weeks ago
• aws-samples / awsome-distributed-training
  Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.
  ☆386Updated this week
• albanD / subclass_zoo
  ☆189Updated last year
• aws-neuron / transformers-neuronx
  ☆111Updated last year
• aws-neuron / neuronx-distributed-inference
  ☆26Updated 2 weeks ago
• aws-neuron / neuronx-distributed-training
  ☆14Updated last month
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆255Updated this week
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark + Toolkit with Torch -> CUDA (+ more DSLs)
  ☆781Updated 2 weeks ago
• aws-samples / sagemaker-trainium-inferentia
  ☆14Updated 2 years ago
• awslabs / slapo
  A schedule language for large model training
  ☆152Updated 5 months ago
• aws-neuron / aws-neuron-parallelcluster-samples
  ☆23Updated 2 months ago
• 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
  ☆43Updated 2 years ago
• aws-samples / ec2-topology-aware-for-slurm
  ☆12Updated 8 months ago
• yifuwang / symm-mem-recipes
  ☆159Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆219Updated last week
• AI-Hypercomputer / jetstream-pytorch
  PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"
  ☆79Updated last month
• facebookexperimental / triton
  Github mirror of trition-lang/triton repo.
  ☆129Updated this week
• perplexityai / pplx-kernels
  Perplexity GPU Kernels
  ☆554Updated 2 months ago
• huggingface / optimum-neuron
  Training and inference on AWS Trainium and Inferentia chips.
  ☆258Updated this week
• aws / aws-ofi-nccl
  This is a plugin which lets EC2 developers use libfabric as network provider while running NCCL applications.
  ☆203Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆277Updated this week
• awslabs / raf
  ☆145Updated last year
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆772Updated last week
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆732Updated this week
• dropbox / gemlite
  Fast low-bit matmul kernels in Triton
  ☆424Updated this week