Skip to content

@aws-neuron AWS Neuron

All

    Repositories list

    32 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 fav…
      Python
      •
      Other
      •202•654•139•2•Updated Nov 13, 2026Nov 13, 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 …
      Python
      •
      Apache License 2.0
      •15•26•1•4•Updated Nov 12, 2026Nov 12, 2026

    • nanochat-contribution-fork

      Public
      The best ChatGPT that $100 can buy.
      Python
      •
      MIT License
      •8.5k•0•0•5•Updated Nov 11, 2026Nov 11, 2026

    • connected-device-maps-over-efa-for-neuron

      Public
      Go
      •
      Apache License 2.0
      •1•0•0•1•Updated Nov 11, 2026Nov 11, 2026

    • aws-neuron-driver

      Public
      Linux kernel device driver supporting AWS Neuron SDK
      C
      •
      GNU General Public License v2.0
      •11•15•1•0•Updated Nov 11, 2026Nov 11, 2026

    • torchtitan-neuron

      Public
      AWS Neuron's extension of TorchTitan that adds AWS Trainium support with TorchNeuron for training generative AI models
      trainingawsmachine-learning
      trainingawsmachine-learningdeep-learningpytorchnkifsdpaws-neurontrainiumtorch-compile
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1.1k•17•0•7•Updated Nov 11, 2026Nov 11, 2026

    • aws-neuron-samples

      Public
      Example code for AWS Neuron SDK developers building inference and training applications
      machine-learningdeep-learningtensorflow
      machine-learningdeep-learningtensorflowpytorch
      Jupyter Notebook
      •
      Other
      •50•163•4•10•Updated Nov 9, 2026Nov 9, 2026

    • torch-neuronx

      Public
      A native PyTorch backend for AWS Trainium with support for eager execution, torch.compile, distributed training, and custom NKI kernels
      trainingawsmachine-learning
      trainingawsmachine-learningdeep-learninginferencepytorchnkifsdpaws-neurontrainium
      Python
      •
      Apache License 2.0
      •11•137•16•1•Updated Oct 30, 2026Oct 30, 2026

    • neuron-helm-charts

      Public
      AWS Neuron Helm Chart repository
      Go Template
      •
      Apache License 2.0
      •2•6•2•1•Updated Oct 28, 2026Oct 28, 2026

    • vllm-omni-neuron

      Public
      vLLM Omni backend plugin for diffusion and multimodal generation on AWS Trainium
      diffusionvllmaws-neuron
      diffusionvllmaws-neurontrainiumvllm-omni
      Python
      •
      Apache License 2.0
      •7•6•0•2•Updated Oct 28, 2026Oct 28, 2026

    • nki-samples

      Public
      Python
      •
      MIT No Attribution
      •43•74•1•1•Updated Oct 26, 2026Oct 26, 2026

    • nkipy

      Public
      NKIPy: Rapid Prototyping on Trainium
      Python
      •
      Apache License 2.0
      •12•30•0•5•Updated Oct 19, 2026Oct 19, 2026

    • nki-library

      Public
      Python
      •
      Apache License 2.0
      •20•78•3•1•Updated Oct 19, 2026Oct 19, 2026

    • neuron-agentic-development

      Public
      Python
      •
      Apache License 2.0
      •10•71•0•3•Updated Oct 19, 2026Oct 19, 2026

    • neuronx-distributed

      Public
      Python
      •
      MIT No Attribution
      •23•69•3•11•Updated Oct 16, 2026Oct 16, 2026

    • neuronx-distributed-inference

      Public
      Python
      •
      Apache License 2.0
      •40•42•2•85•Updated Oct 13, 2026Oct 13, 2026

    • aws-neuron-eks-samples

      Public
      Python
      •
      MIT No Attribution
      •9•25•0•3•Updated Sep 8, 2026Sep 8, 2026

    • upstreaming-to-vllm

      Public archive
      A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      •
      Apache License 2.0
      •25k•24•0•2•Updated Jun 9, 2026Jun 9, 2026

    • nki-moe

      Public
      MLSys competition for the best MOE NKI kernels
      Python
      •
      Apache License 2.0
      •15•48•0•1•Updated May 29, 2026May 29, 2026

    • neuron-workshops

      Public
      Notebooks and sample code for Build On Trainium
      Jupyter Notebook
      •
      Apache License 2.0
      •19•48•7•5•Updated Jan 14, 2026Jan 14, 2026

    • neuronx-distributed-training

      Public
      Python
      •
      Apache License 2.0
      •5•13•0•4•Updated Dec 20, 2025Dec 20, 2025

    • aws-neuron-sagemaker-samples

      Public
      Jupyter Notebook
      •
      MIT No Attribution
      •5•13•0•4•Updated Dec 19, 2025Dec 19, 2025

    • torch2nki

      Public archive
      Project to scale NKI kernels within PyTorch
      Python
      •
      Apache License 2.0
      •2•3•0•0•Updated Dec 3, 2025Dec 3, 2025

    • aws-neuron-parallelcluster-samples

      Public
      Shell
      •
      Other
      •8•24•0•1•Updated Nov 18, 2025Nov 18, 2025

    • nki-llama

      Public
      Project showing how to develop NKI kernels for Llama 3.2 1B inference
      Python
      •
      Apache License 2.0
      •20•22•1•3•Updated May 29, 2025May 29, 2025

    • nki-synthesizer

      Public
      Python
      •
      MIT No Attribution
      •0•0•0•0•Updated Apr 18, 2025Apr 18, 2025

    • transformers-neuronx

      Public
      Python
      •
      Apache License 2.0
      •32•114•0•0•Updated Jan 16, 2025Jan 16, 2025

    • neuronx-nemo-megatron

      Public archive
      Python
      •12•39•0•7•Updated Dec 19, 2024Dec 19, 2024

    • .github

      Public
      507•0•0•0•Updated Oct 18, 2024Oct 18, 2024

    • aws-neuron-reference-for-megatron-lm

      Public
      Python
      •
      Other
      •4•14•0•1•Updated Aug 29, 2023Aug 29, 2023
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.