Accelerator Software Stack
AWS Neuron SDK
awsAWS's compiler and runtime SDK for Trainium and Inferentia, integrating with PyTorch and JAX
Editorial compatibility: these components can be used together in representative deployments. That's not a requirement or a validated implementation. For real deployments, see the sourced reference architectures.