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 501 160

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

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

    Jupyter Notebook 139 39

  3. transformers-neuronx transformers-neuronx Public

    Python 103 28

  4. neuronx-distributed neuronx-distributed Public

    Python 53 10

  5. neuronx-nemo-megatron neuronx-nemo-megatron Public

    Python 35 12

  6. nki-samples nki-samples Public

    Python 28 12

Repositories

Showing 10 of 21 repositories
  • nki-samples Public
    Python 28 MIT-0 12 6 7 Updated Mar 21, 2025
  • 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.

    Python 9 Apache-2.0 11 0 1 Updated Mar 21, 2025
  • upstreaming-to-vllm Public Forked from vllm-project/vllm

    A high-throughput and memory-efficient inference and serving engine for LLMs

    Python 12 Apache-2.0 6,472 0 2 Updated Mar 21, 2025
  • build-on-trainium-workshop Public

    Notebooks and sample code for Build On Trainium

    Jupyter Notebook 4 Apache-2.0 1 8 0 Updated Mar 20, 2025
  • 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 501 160 191 (1 issue needs help) 19 Updated Mar 19, 2025
  • torch2nki Public

    Project to scale NKI kernels within PyTorch

    Python 2 Apache-2.0 2 0 0 Updated Mar 11, 2025
  • Python 53 MIT-0 10 11 10 Updated Feb 12, 2025
  • aws-neuron-samples Public

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

    Jupyter Notebook 139 39 13 11 Updated Feb 11, 2025
  • Python 21 MIT-0 7 1 2 Updated Jan 23, 2025
  • Python 103 Apache-2.0 28 20 6 Updated Jan 16, 2025