aws-neuron / neuronx-nemo-megatronLinks
☆39Updated last year
Alternatives and similar repositories for neuronx-nemo-megatron
Users that are interested in neuronx-nemo-megatron are comparing it to the libraries listed below
Sorting:
- ☆23Updated 2 months ago
- ☆15Updated 4 years ago
- ☆111Updated last year
- ☆64Updated last month
- ☆58Updated last week
- Example code for AWS Neuron SDK developers building inference and training applications☆156Updated 2 weeks ago
- 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
- Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and i…☆573Updated last week
- ☆14Updated 2 years ago
- Training and inference on AWS Trainium and Inferentia chips.☆258Updated this week
- Distributed preprocessing and data loading for language datasets☆40Updated last year
- A high-throughput and memory-efficient inference and serving engine for LLMs☆24Updated 2 months ago
- Collection of best practices, reference architectures, model training examples and utilities to train large models on AWS.☆386Updated this week
- ☆152Updated 3 weeks ago
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆79Updated last month
- A high performance data access library for machine learning tasks☆74Updated 2 years ago
- A CLI tool that helps manage training jobs on the SageMaker HyperPod clusters orchestrated by Amazon EKS☆34Updated last week
- Reference models for Intel(R) Gaudi(R) AI Accelerator☆170Updated 3 weeks ago
- ☆57Updated last week
- Toolkit for allowing inference and serving with PyTorch on SageMaker. Dockerfiles used for building SageMaker Pytorch Containers are at h…☆142Updated last year
- Torch Distributed Experimental☆117Updated last year
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆164Updated 3 weeks ago
- Amazon SageMaker Managed Spot Training Examples☆51Updated last year
- ☆73Updated last year
- SageMaker Studio Docker CLI Extension☆13Updated last year
- ☆192Updated 3 weeks ago
- CLI for building Docker images in SageMaker Studio using AWS CodeBuild.☆58Updated 3 years ago
- ☆72Updated this week
- Create, List, Update, Delete Amazon EKS clusters. Deploy and manage software on EKS. Run distributed model training and inference example…☆64Updated last week
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆205Updated this week