Skip to content
Change the repository type filter

All

    Repositories list

    • nkipy

      Public
      NKIPy: Rapid Prototyping on Trainium
      Python
      41923Updated Feb 24, 2026Feb 24, 2026
    • Python
      82502Updated Feb 18, 2026Feb 18, 2026
    • nki-moe

      Public
      MLSys competition for the best MOE NKI kernels
      Python
      53312Updated Feb 17, 2026Feb 17, 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
      122022Updated Feb 13, 2026Feb 13, 2026
    • Python
      345944Updated Feb 10, 2026Feb 10, 2026
    • Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your fav…
      Python
      1835837325Updated Feb 5, 2026Feb 5, 2026
    • Python
      2230121Updated Feb 2, 2026Feb 2, 2026
    • Python
      24300Updated Jan 24, 2026Jan 24, 2026
    • Example code for AWS Neuron SDK developers building inference and training applications
      Jupyter Notebook
      47158511Updated Jan 15, 2026Jan 15, 2026
    • Notebooks and sample code for Build On Trainium
      Jupyter Notebook
      184674Updated Jan 14, 2026Jan 14, 2026
    • AWS Neuron Helm Chart repository
      Smarty
      1621Updated Dec 29, 2025Dec 29, 2025
    • Python
      2163211Updated Dec 20, 2025Dec 20, 2025
    • Python
      41404Updated Dec 20, 2025Dec 20, 2025
    • Linux kernel device driver supporting AWS Neuron SDK
      C
      81200Updated Dec 20, 2025Dec 20, 2025
    • Jupyter Notebook
      51304Updated Dec 19, 2025Dec 19, 2025
    • torch2nki

      Public archive
      Project to scale NKI kernels within PyTorch
      Python
      2300Updated Dec 3, 2025Dec 3, 2025
    • Shell
      72301Updated Nov 18, 2025Nov 18, 2025
    • A high-throughput and memory-efficient inference and serving engine for LLMs
      Python
      14k2501Updated Nov 7, 2025Nov 7, 2025
    • nki-llama

      Public
      Project showing how to develop NKI kernels for Llama 3.2 1B inference
      Python
      202113Updated May 29, 2025May 29, 2025
    • Python
      0000Updated Apr 18, 2025Apr 18, 2025
    • Python
      3111000Updated Jan 16, 2025Jan 16, 2025
    • neuronx-nemo-megatron

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

      Public
      437000Updated Oct 18, 2024Oct 18, 2024
    • Python
      41401Updated Aug 29, 2023Aug 29, 2023
    • Python
      4500Updated Jun 22, 2023Jun 22, 2023
    • 2600Updated Nov 30, 2022Nov 30, 2022