aws-neuron / neuronx-distributedLinks
☆60Updated this week
Alternatives and similar repositories for neuronx-distributed
Users that are interested in neuronx-distributed are comparing it to the libraries listed below
Sorting:
- ☆40Updated 2 weeks ago
- Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and i…☆533Updated this week
- ☆112Updated 6 months ago
- ☆39Updated 7 months ago
- Example code for AWS Neuron SDK developers building inference and training applications☆148Updated this week
- Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.☆235Updated this week
- ☆14Updated this week
- EFA/NCCL base AMI build Packer and CodeBuild/Pipeline files. Also base Docker build files to enable EFA/NCCL in containers☆44Updated last year
- Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.☆331Updated this week
- ☆23Updated last year
- Project showing how to develop NKI kernels for Llama 3.2 1B inference☆19Updated 2 months ago
- A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS☆27Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆196Updated last week
- ☆12Updated this week
- ☆51Updated last week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆19Updated this week
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆206Updated last week
- Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors☆162Updated last year
- Reference models for Intel(R) Gaudi(R) AI Accelerator☆167Updated last week
- A high performance data access library for machine learning tasks☆74Updated last year
- ☆24Updated 2 months ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆378Updated last week
- The Amazon S3 Connector for PyTorch delivers high throughput for PyTorch training jobs that access and store data in Amazon S3.☆172Updated last week
- Foundation model benchmarking tool. Run any model on any AWS platform and benchmark for performance across instance type and serving stac…☆247Updated 3 months ago
- Hands-on workshop for distributed training and hosting on SageMaker☆144Updated last week
- ☆14Updated 4 years ago
- ☆19Updated this week
- ☆251Updated last year
- A schedule language for large model training☆149Updated last year
- A library to analyze PyTorch traces.☆400Updated last week