Popular repositories Loading
-
aws-neuron-sdk
aws-neuron-sdk PublicPowering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
-
aws-neuron-samples
aws-neuron-samples PublicExample code for AWS Neuron SDK developers building inference and training applications
-
torch-neuronx
torch-neuronx PublicA native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels
-
-
-
Repositories
- aws-neuron-sdk Public
Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
- deep-learning-containers Public
AWS Neuron Deep Learning Containers (DLCs) are a set of Docker images for training and serving models on AWS Trainium and Inferentia instances using AWS Neuron SDK.
-
-
-
- torchtitan-neuron Public Forked from pytorch/torchtitan
AWS Neuron's extension of TorchTitan that adds AWS Trainium support with TorchNeuron for training generative AI models
- aws-neuron-samples Public
Example code for AWS Neuron SDK developers building inference and training applications
- torch-neuronx Public
A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels
-
- vllm-omni-neuron Public
vLLM Omni backend plugin for diffusion and multimodal generation on AWS Trainium
