Alternatives and similar repositories for nki-samples:

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

• aws-neuron / neuronx-distributed
  ☆52Updated last month
• aws-neuron / neuronx-nemo-megatron
  ☆35Updated 2 months ago
• awslabs / nki-autotune
  ☆11Updated this week
• aws-neuron / transformers-neuronx
  ☆102Updated last month
• aws-neuron / aws-neuron-parallelcluster-samples
  ☆23Updated 11 months ago
• awslabs / slapo
  A schedule language for large model training
  ☆145Updated 8 months ago
• aws-neuron / aws-neuron-samples
  Example code for AWS Neuron SDK developers building inference and training applications
  ☆137Updated last month
• cchan / tccl
  extensible collectives library in triton
  ☆83Updated 5 months ago
• stanford-cs149 / asst4-trainium
  ☆23Updated 3 months ago
• aws-samples / aws-parallelcluster-megatron
  ☆14Updated 3 years ago
• awslabs / lorien
  ☆44Updated last year
• awslabs / ratex
  ☆23Updated 3 months ago
• triton-lang / kernels
  ☆73Updated 4 months ago
• NVIDIA / LDDL
  Distributed preprocessing and data loading for language datasets
  ☆39Updated 11 months ago
• stanford-futuredata / stk
  ☆100Updated 6 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
  ☆42Updated last year
• hao-ai-lab / MuxServe
  ☆52Updated 9 months ago
• DachengLi1 / AMP
  (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.
  ☆38Updated 2 years ago
• octoml / octoml-profile
  Home for OctoML PyTorch Profiler
  ☆107Updated last year
• google / iopddl
  Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning
  ☆23Updated 3 months ago
• microsoft / dist-ir
  An IR for efficiently simulating distributed ML computation.
  ☆28Updated last year
• aws-samples / awsome-inference
  ☆43Updated last month
• RulinShao / FastCkpt
  Python package for rematerialization-aware gradient checkpointing
  ☆24Updated last year
• ScalingIntelligence / KernelBench
  KernelBench: Can LLMs Write GPU Kernels? - Benchmark with Torch -> CUDA problems
  ☆227Updated last week
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆62Updated last year
• SymbioticLab / Oobleck
  A resilient distributed training framework
  ☆89Updated 11 months ago
• huggingface / optimum-neuron
  Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.
  ☆219Updated this week
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆56Updated this week