Alternatives and similar repositories for nki-samples:

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

• awslabs / nki-autotune
  ☆11Updated this week
• aws-neuron / neuronx-distributed
  ☆51Updated this week
• aws-neuron / neuronx-nemo-megatron
  ☆34Updated last month
• awslabs / ratex
  ☆23Updated 2 months ago
• awslabs / slapo
  A schedule language for large model training
  ☆144Updated 8 months ago
• stanford-cs149 / asst4-trainium
  ☆23Updated 2 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆82Updated 4 months ago
• aws-neuron / aws-neuron-parallelcluster-samples
  ☆23Updated 10 months ago
• triton-lang / kernels
  ☆67Updated 3 months ago
• hao-ai-lab / MuxServe
  ☆50Updated 8 months ago
• CentML / DeepView.Profile
  🏙 Interactive performance profiling and debugging tool for PyTorch neural networks.
  ☆58Updated 3 weeks ago
• aws-neuron / aws-neuron-samples
  Example code for AWS Neuron SDK developers building inference and training applications
  ☆135Updated last week
• octoml / octoml-profile
  Home for OctoML PyTorch Profiler
  ☆107Updated last year
• aws-samples / aws-parallelcluster-megatron
  ☆14Updated 3 years ago
• zhisbug / Cavs
  Cavs: An Efficient Runtime System for Dynamic Neural Networks
  ☆14Updated 4 years 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
  ☆42Updated last year
• awslabs / lorien
  ☆44Updated last year
• NVIDIA / LDDL
  Distributed preprocessing and data loading for language datasets
  ☆39Updated 10 months ago
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 2 months ago
• aws-neuron / transformers-neuronx
  ☆102Updated last month
• stanford-futuredata / stk
  ☆99Updated 5 months ago
• Jokeren / triton-samples
  ☆24Updated last month
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆166Updated last week
• ucamrl / xrlflow
  ☆14Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆187Updated this week
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆38Updated 9 months ago
• UofT-EcoSystem / BPPSA-open
  The (open-source part of) code to reproduce "BPPSA: Scaling Back-propagation by Parallel Scan Algorithm".
  ☆13Updated 3 years ago
• illinois-impact / klap
  A source-to-source compiler for optimizing CUDA dynamic parallelism by aggregating launches
  ☆15Updated 5 years ago
• tonyzhang617 / nomad-dist
  ☆26Updated 11 months ago
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆27Updated last year