This document is relevant for: Inf2, Trn1, Trn1n

API Reference Guide (tensorflow-neuronx)

• TensorFlow 2.x (tensorflow-neuronx) Tracing API

• TensorFlow Neuron (tensorflow-neuronx) Auto Multicore Replication (Beta)

• TensorFlow 2.x (tensorflow-neuronx) analyze_model API

This document is relevant for: Inf2, Trn1, Trn1n