aws-neuron / neuronx-distributedLinks
☆60Updated 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:
- Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and i…☆548Updated last week
- ☆51Updated 3 weeks ago
- ☆110Updated 9 months ago
- Example code for AWS Neuron SDK developers building inference and training applications☆149Updated last week
- ☆39Updated 10 months ago
- Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.☆245Updated this week
- ☆23Updated last year
- Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.☆355Updated last week
- ☆15Updated this week
- 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
- Project showing how to develop NKI kernels for Llama 3.2 1B inference☆19Updated 4 months ago
- Amazon SageMaker Debugger provides functionality to save tensors during training of machine learning jobs and analyze those tensors☆162Updated last year
- ☆56Updated last month
- ☆14Updated last year
- A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS☆32Updated last week
- ☆24Updated 4 months ago
- ☆14Updated 4 years ago
- A high performance data access library for machine learning tasks☆74Updated last year
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆215Updated this week
- ☆21Updated last month
- ☆13Updated last month
- A high-throughput and memory-efficient inference and serving engine for LLMs☆21Updated last week
- Hands-on workshop for distributed training and hosting on SageMaker☆148Updated last week
- Toolkit for allowing inference and serving with PyTorch on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at h…☆140Updated last year
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆226Updated last week
- ☆252Updated last year
- ☆72Updated last week
- Distributed preprocessing and data loading for language datasets☆39Updated last year
- A helper library to connect into Amazon SageMaker with AWS Systems Manager and SSH (Secure Shell)☆255Updated 3 months ago
- Neo-AI-DLR is a common runtime for machine learning models compiled by AWS SageMaker Neo, TVM, or TreeLite.☆497Updated 2 years ago