Skip to content

aws-neuron/aws-neuron-sdk

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1,129 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

neuron

AWS Neuron

Neuron SDK Overview

AWS Neuron is a software development kit (SDK) enabling high-performance deep learning and generative AI using AWS Inferentia and Trainium accelerators. With Neuron, you can develop, profile, and deploy machine learning workloads on accelerated EC2 instances (Inf1, Inf2, Trn1, Trn2, Trn3).

Neuron includes a graph compiler, runtime, developer tools (Neuron Explorer, agentic development tools), and integrates with PyTorch, JAX, and vLLM Neuron for inference serving. The Neuron Kernel Interface (NKI) provides direct NeuronCore programming for custom kernels.

Neuron SDK's documentation

For full documentation including user guides, tutorials, and API references, see Neuron SDK documentation.

Support

If none of the GitHub and online resources have an answer to your question, visit AWS re:Post for community support.

About

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

Resources

License

Unknown and 2 other licenses found

Licenses found

Unknown
LICENSE-DOCUMENTATION
MIT-0
LICENSE-SAMPLECODE
Unknown
LICENSE-SUMMARY-DOCS-SAMPLES

Code of conduct

Contributing

Security policy

Stars

616 stars

Watchers

88 watching

Forks

Packages

 
 
 

Contributors