TensorFlow Neuron

Neuron is integrated into TensorFlow, and provides you with a familiar environment to run inference using Inferentia based instances.

Neuron supports both Tensorflow 2.x, an eager-execution-based deep learning framework, and TensorFlow 1.x, a static-graph-based deep learning framework. Currently, the supported TensorFlow versions are 2.5.0, 2.4.2, 2.3.3, 2.2.3, 2.1.4, and 1.15.5.