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
  ☆111Updated 11 months ago
• aws-neuron / nki-samples
  ☆58Updated last 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…
  ☆569Updated this week
• aws-neuron / aws-neuron-samples
  Example code for AWS Neuron SDK developers building inference and training applications
  ☆152Updated last week
• aws-neuron / neuronx-nemo-megatron
  ☆39Updated last year
• huggingface / optimum-neuron
  Training and inference on AWS Trainium and Inferentia chips.
  ☆253Updated 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
  ☆43Updated 2 years ago
• aws-neuron / aws-neuron-parallelcluster-samples
  ☆23Updated last month
• aws-samples / awsome-distributed-training
  Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.
  ☆380Updated this week
• aws-samples / awsome-inference
  ☆57Updated 2 weeks ago
• awslabs / sagemaker-debugger
  Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors
  ☆162Updated last year
• 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.
  ☆195Updated last week
• awslabs / nki-autotune
  ☆15Updated last week
• HabanaAI / Model-References
  Reference models for Intel(R) Gaudi(R) AI Accelerator
  ☆169Updated 3 months ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆216Updated 2 weeks ago
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆239Updated this week
• aws-neuron / upstreaming-to-vllm
  A high-throughput and memory-efficient inference and serving engine for LLMs
  ☆25Updated last month
• aws-neuron / aws-neuron-eks-samples
  ☆25Updated 3 weeks ago
• aws-neuron / nki-llama
  Project showing how to develop NKI kernels for Llama 3.2 1B inference
  ☆20Updated 6 months ago
• aws / sagemaker-hyperpod-cli
  A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS
  ☆34Updated last week
• octoml / octoml-profile
  Home for OctoML PyTorch Profiler
  ☆114Updated 2 years ago
• AI-Hypercomputer / JetStream
  JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…
  ☆396Updated 6 months ago
• microsoft / varuna
  ☆252Updated last year
• aws-samples / aws-parallelcluster-megatron
  ☆15Updated 4 years ago
• NVIDIA / LDDL
  Distributed preprocessing and data loading for language datasets
  ☆40Updated last year
• aws / aws-ofi-nccl
  This is a plugin which lets EC2 developers use libfabric as network provider while running NCCL applications.
  ☆202Updated last week
• aws-neuron / neuronx-distributed-training
  ☆13Updated last week
• huggingface / optimum-habana
  Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)
  ☆203Updated this week
• awslabs / ml-io
  A high performance data access library for machine learning tasks
  ☆74Updated 2 years ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆459Updated 3 weeks ago