Inference Stack
Back to the interactive map
Hardware Abstraction

AWS Neuron SDK

aws

AWS's compiler and runtime SDK for Trainium and Inferentia, integrating with PyTorch and JAX

Official docs

Used by

Leads to