This document is relevant for: Inf1, Inf2, Trn1, Trn1n

Neuron Compiler

The Neuron Compiler accepts Machine Learning models in various formats (TensorFlow, MXNet, PyTorch, XLA HLO) and optimizes them to run on Neuron devices.

The Neuron compiler is invoked within the ML framework, where ML models are sent to the compiler by the Neuron Framework plugin. The resulting compiler artifact is called a NEFF file (Neuron Executable File Format) that in turn is loaded by the Neuron runtime to the Neuron device.

Neuron Compiler for Trn1 & Inf2

API Reference Guide

• Neuron Compiler CLI Reference Guide

Developer Guide

• Mixed Precision and Performance-accuracy Tuning (neuronx-cc)

Misc

• FAQ

• What’s New

Neuron Compiler for Inf1

API Reference Guide

• Neuron compiler CLI Reference Guide (neuron-cc)

Developer Guide

• Mixed precision and performance-accuracy tuning (neuron-cc)

Misc

• FAQ

• What’s New

• Neuron Supported operators

This document is relevant for: Inf1, Inf2, Trn1, Trn1n