Use SageMaker with Trainium and Inferentia for optimized deep learning training and inferencing workloads
This work demonstrates how to use SageMaker to leverage AWS Trainium and AWS Inferentia for an end-to-end experience of model training and inferencing. Both Trainium and Inferentia processors are designed and optimized to support deep learning workloads in AWS Cloud.
Clone this repo in your SageMaker instance, and then start with this notebook to learn how to use SageMaker to leverage these AWS processors.
This library is licensed under the Apache 2.0 License. For more details, please take a look at the LICENSE file.