AWS Trainium es un acelerador de aprendizaje automático (AM) personalizado diseñado por Amazon Web Services (AWS) para optimizar el entrenamiento de los modelos de aprendizaje automático.
Forma parte de la estrategia más amplia de AWS de proporcionar hardware de alto rendimiento y rentable para cargas de trabajo de ML, que permita a los clientes entrenar modelos grandes y complejos más rápidamente y a un coste inferior al de los procesadores tradicionales.
Trainium se ha optimizado para entrenar modelos de procesamiento del lenguaje natural, visión por ordenador y recomendación utilizados en un amplio conjunto de aplicaciones, como resumen de textos, generación de códigos, respuesta a preguntas, generación de imágenes y vídeos, recomendación y detección de fraudes.