Skip to content

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 583 183

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

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

    Jupyter Notebook 158 47

  3. transformers-neuronx transformers-neuronx Public

    Python 110 31

  4. neuronx-distributed neuronx-distributed Public

    Python 63 21

  5. nki-samples nki-samples Public

    Python 59 34

  6. neuron-workshops neuron-workshops Public

    Notebooks and sample code for Build On Trainium

    Jupyter Notebook 47 18

Repositories

Showing 10 of 26 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

    aws-neuron/aws-neuron-sdk’s past year of commit activity
    Python 583 183 73 25 Updated Mar 3, 2026
  • nkipy Public

    NKIPy: Rapid Prototyping on Trainium

    aws-neuron/nkipy’s past year of commit activity
    Python 22 Apache-2.0 5 2 2 Updated Mar 1, 2026
  • 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 1 2 1 Updated Feb 28, 2026
  • nki-library Public
    aws-neuron/nki-library’s past year of commit activity
    Python 45 Apache-2.0 2 0 0 Updated Feb 27, 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 21 Apache-2.0 12 2 2 Updated Feb 27, 2026
  • aws-neuron/aws-neuron-eks-samples’s past year of commit activity
    Python 25 MIT-0 8 0 2 Updated Feb 26, 2026
  • aws-neuron-driver Public

    Linux kernel device driver supporting AWS Neuron SDK

    aws-neuron/aws-neuron-driver’s past year of commit activity
    C 13 GPL-2.0 8 0 0 Updated Feb 26, 2026
  • nki-moe Public

    MLSys competition for the best MOE NKI kernels

    aws-neuron/nki-moe’s past year of commit activity
    Python 36 Apache-2.0 6 1 2 Updated Feb 17, 2026
  • nki-samples Public
    aws-neuron/nki-samples’s past year of commit activity
    Python 59 MIT-0 34 3 4 Updated Feb 10, 2026
  • aws-neuron/neuronx-distributed-inference’s past year of commit activity
    Python 31 Apache-2.0 22 1 22 Updated Feb 2, 2026

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Most used topics

Loading…