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 595 185

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

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

    Jupyter Notebook 158 48

  3. transformers-neuronx transformers-neuronx Public

    Python 111 31

  4. neuronx-distributed neuronx-distributed Public

    Python 64 21

  5. nki-samples nki-samples Public

    Python 62 36

  6. nki-library nki-library Public

    Python 53 5

Repositories

Showing 10 of 27 repositories
  • nkipy Public

    NKIPy: Rapid Prototyping on Trainium

    aws-neuron/nkipy’s past year of commit activity
    Python 26 Apache-2.0 9 2 6 Updated Apr 20, 2026
  • nki-moe Public

    MLSys competition for the best MOE NKI kernels

    aws-neuron/nki-moe’s past year of commit activity
    Python 42 Apache-2.0 9 0 0 Updated Apr 20, 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 595 185 76 23 Updated Apr 20, 2026
  • neuronx-distributed-inference Public
    aws-neuron/neuronx-distributed-inference’s past year of commit activity
    Python 34 Apache-2.0 28 1 64 Updated Apr 17, 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 Apr 17, 2026
  • nki-samples Public
    aws-neuron/nki-samples’s past year of commit activity
    Python 62 MIT-0 36 2 6 Updated Apr 16, 2026
  • nki-library Public
    aws-neuron/nki-library’s past year of commit activity
    Python 53 Apache-2.0 5 0 0 Updated Apr 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 SDK.

    aws-neuron/deep-learning-containers’s past year of commit activity
    Python 21 Apache-2.0 12 1 2 Updated Apr 10, 2026
  • neuronx-distributed Public
    aws-neuron/neuronx-distributed’s past year of commit activity
    Python 64 MIT-0 21 2 11 Updated Apr 9, 2026
  • connected-device-maps-over-efa-for-neuron Public
    aws-neuron/connected-device-maps-over-efa-for-neuron’s past year of commit activity
    Go 0 Apache-2.0 0 0 0 Updated Apr 9, 2026
View all repositories

People

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

Top languages

Loading…

Most used topics

Loading…