Skip to content
@aws-neuron

AWS Neuron

Popular repositories Loading

  1. aws-neuron-sdk 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

    Python 649 200

  2. aws-neuron-samples aws-neuron-samples Public

    Example code for AWS Neuron SDK developers building inference and training applications

    Jupyter Notebook 163 51

  3. torch-neuronx torch-neuronx Public

    A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels

    Python 113 6

  4. transformers-neuronx transformers-neuronx Public

    Python 112 32

  5. nki-library nki-library Public

    Python 73 16

  6. nki-samples nki-samples Public

    Python 72 43

Repositories

Showing 10 of 32 repositories
  • neuron-helm-charts Public

    AWS Neuron Helm Chart repository

    aws-neuron/neuron-helm-charts's past year of commit activity
    Go Template 6 Apache-2.0 2 2 1 Updated Oct 21, 2026
  • 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

    aws-neuron/aws-neuron-sdk's past year of commit activity
    Python 649 200 95 2 Updated Oct 21, 2026
  • nanochat-contribution-fork Public Forked from karpathy/nanochat

    The best ChatGPT that $100 can buy.

    aws-neuron/nanochat-contribution-fork's past year of commit activity
    Python 0 MIT 8,307 0 3 Updated Oct 21, 2026
  • torch-neuronx Public

    A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels

    aws-neuron/torch-neuronx's past year of commit activity
    Python 113 Apache-2.0 6 3 0 Updated Oct 20, 2026
  • 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.

    aws-neuron/deep-learning-containers's past year of commit activity
    Python 25 Apache-2.0 14 1 5 Updated Oct 20, 2026
  • torchtitan-neuron Public Forked from pytorch/torchtitan

    Fork of PyTorch TorchTitan that adds AWS Trainium support with TorchNeuron

    aws-neuron/torchtitan-neuron's past year of commit activity
    Python 14 BSD-3-Clause 1,063 0 6 Updated Oct 20, 2026
  • nkipy Public

    NKIPy: Rapid Prototyping on Trainium

    aws-neuron/nkipy's past year of commit activity
    Python 30 Apache-2.0 12 0 5 Updated Oct 19, 2026
  • nki-library Public
    aws-neuron/nki-library's past year of commit activity
    Python 73 Apache-2.0 16 1 6 Updated Oct 19, 2026
  • neuron-agentic-development Public
    aws-neuron/neuron-agentic-development's past year of commit activity
    Python 63 Apache-2.0 10 0 3 Updated Oct 19, 2026
  • vllm-omni-neuron Public

    vLLM Omni backend plugin for diffusion and multimodal generation on AWS Trainium

    aws-neuron/vllm-omni-neuron's past year of commit activity
    Python 3 Apache-2.0 6 0 1 Updated Oct 20, 2026
View all repositories

Top languages

Loading…