Select your cookie preferences

We use essential cookies and similar tools that are necessary to provide our site and services. We use performance cookies to collect anonymous statistics, so we can understand how customers use our site and make improvements. Essential cookies cannot be deactivated, but you can choose “Customize” or “Decline” to decline performance cookies.

If you agree, AWS and approved third parties will also use cookies to provide useful site features, remember your preferences, and display relevant content, including relevant advertising. To accept or decline all non-essential cookies, choose “Accept” or “Decline.” To make more detailed choices, choose “Customize.”

MLCOST-12: Select an optimal ML framework - Machine Learning Lens

MLCOST-12: Select an optimal ML framework

Organize, track, compare and evaluate machine learning (ML) experiments and model versions. Identify the most cost-effective and optimal combination of instance types and ML frameworks. Examples of ML frameworks include TensorFlow, PyTorch, and Scikit-learn.

Implementation plan

  • Use Amazon SageMaker AI Experiments - Amazon SageMaker AI Experiments lets you organize, track, compare, and evaluate your machine learning experiments. Using this service, you can experiment with various ML frameworks and see which one gives you the most cost-effective performance. AWS Deep Learning AMIs and AWS Deep Learning Containers enable you to use several open-source ML frameworks for training on your infrastructure. AWS Deep Learning AMIss have popular deep learning frameworks and interfaces preinstalled including TensorFlow, PyTorch, Apache MXNet, Chainer, Gluon, Horovod, and Keras. The AMI or container can be launched on powerful infrastructure that has been optimized for ML performance. SageMaker AI also allows you to bring your own container where you can use any framework you choose.

Documents

Blogs

PrivacySite termsCookie preferences
© 2025, Amazon Web Services, Inc. or its affiliates. All rights reserved.