aws-neuron / neuronx-distributedLinks
☆63Updated last month
Alternatives and similar repositories for neuronx-distributed
Users that are interested in neuronx-distributed are comparing it to the libraries listed below
Sorting:
- ☆53Updated this week
- ☆111Updated 10 months ago
- Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and i…☆553Updated this week
- ☆39Updated 10 months ago
- Example code for AWS Neuron SDK developers building inference and training applications☆150Updated this week
- Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.☆366Updated this week
- Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.☆247Updated last week
- ☆23Updated last year
- 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
- ☆15Updated this week
- ☆56Updated 2 weeks ago
- Reference models for Intel(R) Gaudi(R) AI Accelerator☆166Updated last month
- The Amazon S3 Connector for PyTorch delivers high throughput for PyTorch training jobs that access and store data in Amazon S3.☆190Updated 2 weeks ago
- Project showing how to develop NKI kernels for Llama 3.2 1B inference☆19Updated 5 months ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆21Updated last week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆232Updated last week
- ☆24Updated 5 months ago
- ☆15Updated 4 years ago
- A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS☆32Updated this week
- ☆13Updated last month
- ☆14Updated last year
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆216Updated this week
- This is a plugin which lets EC2 developers use libfabric as network provider while running NCCL applications.☆192Updated this week
- ☆22Updated last month
- Applied AI experiments and examples for PyTorch☆303Updated 2 months ago
- Distributed preprocessing and data loading for language datasets☆39Updated last year
- ☆252Updated last year
- Pipeline Parallelism for PyTorch☆784Updated last year
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆400Updated this week
- JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs wel…☆388Updated 5 months ago