Skip to content
Change the repository type filter

All

    Repositories list

    • 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
      1855967623Updated Apr 22, 2026Apr 22, 2026
    • nkipy

      Public
      NKIPy: Rapid Prototyping on Trainium
      Python
      Apache License 2.0
      92625Updated Apr 21, 2026Apr 21, 2026
    • Python
      Apache License 2.0
      2834156Updated Apr 21, 2026Apr 21, 2026
    • nki-moe

      Public
      MLSys competition for the best MOE NKI kernels
      Python
      Apache License 2.0
      94200Updated Apr 20, 2026Apr 20, 2026
    • AWS Neuron Helm Chart repository
      Go Template
      Apache License 2.0
      1621Updated Apr 17, 2026Apr 17, 2026
    • nki-samples

      Public
      Python
      MIT No Attribution
      366326Updated Apr 16, 2026Apr 16, 2026
    • nki-library

      Public
      Python
      Apache License 2.0
      55300Updated Apr 13, 2026Apr 13, 2026
    • 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
      122112Updated Apr 10, 2026Apr 10, 2026
    • Python
      MIT No Attribution
      2164211Updated Apr 9, 2026Apr 9, 2026
    • Go
      Apache License 2.0
      0000Updated Apr 9, 2026Apr 9, 2026
    • Linux kernel device driver supporting AWS Neuron SDK
      C
      GNU General Public License v2.0
      101300Updated Apr 9, 2026Apr 9, 2026
    • Example code for AWS Neuron SDK developers building inference and training applications
      Jupyter Notebook
      Other
      48158511Updated Apr 2, 2026Apr 2, 2026
    • Python
      MIT No Attribution
      82401Updated Mar 30, 2026Mar 30, 2026
    • A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      Apache License 2.0
      16k2502Updated Mar 5, 2026Mar 5, 2026
    • Notebooks and sample code for Build On Trainium
      Jupyter Notebook
      Apache License 2.0
      184774Updated Jan 14, 2026Jan 14, 2026
    • Python
      Apache License 2.0
      41204Updated Dec 20, 2025Dec 20, 2025
    • Jupyter Notebook
      MIT No Attribution
      51304Updated Dec 19, 2025Dec 19, 2025
    • torch2nki

      Public archive
      Project to scale NKI kernels within PyTorch
      Python
      Apache License 2.0
      2300Updated Dec 3, 2025Dec 3, 2025
    • aws-neuron-parallelcluster-samples

      Public
      Shell
      Other
      72401Updated 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
      202113Updated May 29, 2025May 29, 2025
    • nki-synthesizer

      Public
      Python
      MIT No Attribution
      0000Updated Apr 18, 2025Apr 18, 2025
    • Python
      Apache License 2.0
      3111100Updated Jan 16, 2025Jan 16, 2025
    • neuronx-nemo-megatron

      Public archive
      Python
      123907Updated Dec 19, 2024Dec 19, 2024
    • .github

      Public
      449000Updated Oct 18, 2024Oct 18, 2024
    • Python
      Other
      41401Updated Aug 29, 2023Aug 29, 2023
    • Python
      Apache License 2.0
      4500Updated Jun 22, 2023Jun 22, 2023
    • Apache License 2.0
      2600Updated Nov 30, 2022Nov 30, 2022
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.