Alternatives and similar repositories for neuronx-distributed

Users that are interested in neuronx-distributed are comparing it to the libraries listed below

• aws-neuron / transformers-neuronx
  ☆108Updated 4 months ago
• aws-neuron / aws-neuron-samples
  Example code for AWS Neuron SDK developers building inference and training applications
  ☆146Updated this week
• aws-neuron / nki-samples
  ☆35Updated last week
• huggingface / optimum-neuron
  Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.
  ☆227Updated last week
• aws-neuron / neuronx-distributed-training
  ☆11Updated 2 weeks ago
• aws-neuron / aws-neuron-parallelcluster-samples
  ☆24Updated last year
• aws-neuron / neuronx-nemo-megatron
  ☆35Updated 5 months ago
• awslabs / nki-autotune
  ☆14Updated this week
• 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…
  ☆518Updated this week
• aws-neuron / upstreaming-to-vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆16Updated last week
• aws-neuron / neuronx-distributed-inference
  ☆16Updated 2 weeks ago
• aws-samples / awsome-inference
  ☆47Updated last month
• aws-neuron / aws-neuron-eks-samples
  ☆24Updated this week
• aws-neuron / nki-llama
  Project showing how to develop NKI kernels for Llama 3.2 1B inference
  ☆14Updated last 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
  ☆43Updated last year
• aws-samples / awsome-distributed-training
  Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.
  ☆300Updated this week
• awslabs / s3-connector-for-pytorch
  The Amazon S3 Connector for PyTorch delivers high throughput for PyTorch training jobs that access and store data in Amazon S3.
  ☆165Updated this week
• awslabs / ml-io
  A high performance data access library for machine learning tasks
  ☆74Updated last year
• aws-samples / aws-do-eks
  Create, List, Update, Delete Amazon EKS clusters. Deploy and manage software on EKS. Run distributed model training and inference example…
  ☆59Updated last week
• aws-samples / foundation-model-benchmarking-tool
  Foundation model benchmarking tool. Run any model on any AWS platform and benchmark for performance across instance type and serving stac…
  ☆241Updated last month
• aws-samples / sagemaker-genai-hosting-examples
  ☆102Updated this week
• aws / aws-ofi-nccl
  This is a plugin which lets EC2 developers use libfabric as network provider while running NCCL applications.
  ☆173Updated this week
• aws-solutions-library-samples / guidance-for-machine-learning-inference-on-aws
  This Guidance demonstrates how to deploy a machine learning inference architecture on Amazon Elastic Kubernetes Service (Amazon EKS). It …
  ☆43Updated last week
• aws / sagemaker-hyperpod-cli
  A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS
  ☆21Updated last week
• aws-samples / ec2-topology-aware-for-slurm
  ☆12Updated this week
• aws / sagemaker-hyperpod-recipes
  ☆51Updated 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 …
  ☆169Updated last week
• neo-ai / tvm
  Open deep learning compiler stack for cpu, gpu and specialized accelerators
  ☆89Updated 2 years ago
• awslabs / slapo
  A schedule language for large model training
  ☆148Updated 11 months ago
• microsoft / varuna
  ☆250Updated 10 months ago