Best Machine Learning Software for Amazon SageMaker
View:
Compare the Top Machine Learning Software that integrates with Amazon SageMaker as of June 2025
Sort By:
This a list of Machine Learning software that integrates with Amazon SageMaker. Use the filters on the left to add additional filters for products that have integrations with Amazon SageMaker. View the products that work with Amazon SageMaker in the table below.
What is Machine Learning Software for Amazon SageMaker?
Machine learning software enables developers and data scientists to build, train, and deploy models that can learn from data and make predictions or decisions without being explicitly programmed. These tools provide frameworks and algorithms for tasks such as classification, regression, clustering, and natural language processing. They often come with features like data preprocessing, model evaluation, and hyperparameter tuning, which help optimize the performance of machine learning models. With the ability to analyze large datasets and uncover patterns, machine learning software is widely used in industries like healthcare, finance, marketing, and autonomous systems. Overall, this software empowers organizations to leverage data for smarter decision-making and automation. Compare and read user reviews of the best Machine Learning software for Amazon SageMaker currently available using the table below. This list is updated regularly.
-
1
Domino Enterprise MLOps Platform
Domino Data Lab
The Domino platform helps data science teams improve the speed, quality, and impact of data science at scale. Domino is open and flexible, empowering professional data scientists to use their preferred tools and infrastructure. Data science models get into production fast and are kept operating at peak performance with integrated workflows. Domino also delivers the security, governance and compliance that enterprises expect. The Self-Service Infrastructure Portal makes data science teams become more productive with easy access to their preferred tools, scalable compute, and diverse data sets. The Integrated Model Factory includes a workbench, model and app deployment, and integrated monitoring to rapidly experiment, deploy the best models in production, ensure optimal performance, and collaborate across the end-to-end data science lifecycle. The System of Record allows teams to easily find, reuse, reproduce, and build on any data science work to amplify innovation. -
2
Dataiku
Dataiku
Dataiku is an advanced data science and machine learning platform designed to enable teams to build, deploy, and manage AI and analytics projects at scale. It empowers users, from data scientists to business analysts, to collaboratively create data pipelines, develop machine learning models, and prepare data using both visual and coding interfaces. Dataiku supports the entire AI lifecycle, offering tools for data preparation, model training, deployment, and monitoring. The platform also includes integrations for advanced capabilities like generative AI, helping organizations innovate and deploy AI solutions across industries. -
3
Ray
Anyscale
Develop on your laptop and then scale the same Python code elastically across hundreds of nodes or GPUs on any cloud, with no changes. Ray translates existing Python concepts to the distributed setting, allowing any serial application to be easily parallelized with minimal code changes. Easily scale compute-heavy machine learning workloads like deep learning, model serving, and hyperparameter tuning with a strong ecosystem of distributed libraries. Scale existing workloads (for eg. Pytorch) on Ray with minimal effort by tapping into integrations. Native Ray libraries, such as Ray Tune and Ray Serve, lower the effort to scale the most compute-intensive machine learning workloads, such as hyperparameter tuning, training deep learning models, and reinforcement learning. For example, get started with distributed hyperparameter tuning in just 10 lines of code. Creating distributed apps is hard. Ray handles all aspects of distributed execution.Starting Price: Free -
4
Union Cloud
Union.ai
Union.ai is an award-winning, Flyte-based data and ML orchestrator for scalable, reproducible ML pipelines. With Union.ai, you can write your code locally and easily deploy pipelines to remote Kubernetes clusters. “Flyte’s scalability, data lineage, and caching capabilities enable us to train hundreds of models on petabytes of geospatial data, giving us an edge in our business.” — Arno, CTO at Blackshark.ai “With Flyte, we want to give the power back to biologists. We want to stand up something that they can play around with different parameters for their models because not every … parameter is fixed. We want to make sure we are giving them the power to run the analyses.” — Krishna Yeramsetty, Principal Data Scientist at Infinome “Flyte plays a vital role as a key component of Gojek's ML Platform by providing exactly that." — Pradithya Aria Pura, Principal Engineer at GojStarting Price: Free (Flyte) -
5
NVIDIA Triton™ inference server delivers fast and scalable AI in production. Open-source inference serving software, Triton inference server streamlines AI inference by enabling teams deploy trained AI models from any framework (TensorFlow, NVIDIA TensorRT®, PyTorch, ONNX, XGBoost, Python, custom and more on any GPU- or CPU-based infrastructure (cloud, data center, or edge). Triton runs models concurrently on GPUs to maximize throughput and utilization, supports x86 and ARM CPU-based inferencing, and offers features like dynamic batching, model analyzer, model ensemble, and audio streaming. Triton helps developers deliver high-performance inference aTriton integrates with Kubernetes for orchestration and scaling, exports Prometheus metrics for monitoring, supports live model updates, and can be used in all major public cloud machine learning (ML) and managed Kubernetes platforms. Triton helps standardize model deployment in production.Starting Price: Free
-
6
BentoML
BentoML
Serve your ML model in any cloud in minutes. Unified model packaging format enabling both online and offline serving on any platform. 100x the throughput of your regular flask-based model server, thanks to our advanced micro-batching mechanism. Deliver high-quality prediction services that speak the DevOps language and integrate perfectly with common infrastructure tools. Unified format for deployment. High-performance model serving. DevOps best practices baked in. The service uses the BERT model trained with the TensorFlow framework to predict movie reviews' sentiment. DevOps-free BentoML workflow, from prediction service registry, deployment automation, to endpoint monitoring, all configured automatically for your team. A solid foundation for running serious ML workloads in production. Keep all your team's models, deployments, and changes highly visible and control access via SSO, RBAC, client authentication, and auditing logs.Starting Price: Free -
7
Flyte
Union.ai
The workflow automation platform for complex, mission-critical data and ML processes at scale. Flyte makes it easy to create concurrent, scalable, and maintainable workflows for machine learning and data processing. Flyte is used in production at Lyft, Spotify, Freenome, and others. At Lyft, Flyte has been serving production model training and data processing for over four years, becoming the de-facto platform for teams like pricing, locations, ETA, mapping, autonomous, and more. In fact, Flyte manages over 10,000 unique workflows at Lyft, totaling over 1,000,000 executions every month, 20 million tasks, and 40 million containers. Flyte has been battle-tested at Lyft, Spotify, Freenome, and others. It is entirely open-source with an Apache 2.0 license under the Linux Foundation with a cross-industry overseeing committee. Configuring machine learning and data workflows can get complex and error-prone with YAML.Starting Price: Free -
8
neptune.ai
neptune.ai
Neptune.ai is a machine learning operations (MLOps) platform designed to streamline the tracking, organizing, and sharing of experiments and model-building processes. It provides a comprehensive environment for data scientists and machine learning engineers to log, visualize, and compare model training runs, datasets, hyperparameters, and metrics in real-time. Neptune.ai integrates easily with popular machine learning libraries, enabling teams to efficiently manage both research and production workflows. With features that support collaboration, versioning, and experiment reproducibility, Neptune.ai enhances productivity and helps ensure that machine learning projects are transparent and well-documented across their lifecycle.Starting Price: $49 per month -
9
JFrog ML
JFrog
JFrog ML (formerly Qwak) offers an MLOps platform designed to accelerate the development, deployment, and monitoring of machine learning and AI applications at scale. The platform enables organizations to manage the entire lifecycle of machine learning models, from training to deployment, with tools for model versioning, monitoring, and performance tracking. It supports a wide variety of AI models, including generative AI and LLMs (Large Language Models), and provides an intuitive interface for managing prompts, workflows, and feature engineering. JFrog ML helps businesses streamline their ML operations and scale AI applications efficiently, with integrated support for cloud environments. -
10
Comet
Comet
Manage and optimize models across the entire ML lifecycle, from experiment tracking to monitoring models in production. Achieve your goals faster with the platform built to meet the intense demands of enterprise teams deploying ML at scale. Supports your deployment strategy whether it’s private cloud, on-premise servers, or hybrid. Add two lines of code to your notebook or script and start tracking your experiments. Works wherever you run your code, with any machine learning library, and for any machine learning task. Easily compare experiments—code, hyperparameters, metrics, predictions, dependencies, system metrics, and more—to understand differences in model performance. Monitor your models during every step from training to production. Get alerts when something is amiss, and debug your models to address the issue. Increase productivity, collaboration, and visibility across all teams and stakeholders.Starting Price: $179 per user per month -
11
Superwise
Superwise
Get in minutes what used to take years to build. Simple, customizable, scalable, secure, ML monitoring. Everything you need to deploy, maintain and improve ML in production. Superwise is an open platform that integrates with any ML stack and connects to your choice of communication tools. Want to take it further? Superwise is API-first and everything (and we mean everything) is accessible via our APIs. All from the comfort of the cloud of your choice. When it comes to ML monitoring you have full self-service control over everything. Configure metrics and policies through our APIs and SDK or simply select a monitoring template and set the sensitivity, conditions, and alert channels of your choice. Try Superwise out or contact us to learn more. Easily create alerts with Superwise’s ML monitoring policy templates and builder. Select from dozens of pre-build monitors ranging from data drift to equal opportunity, or customize policies to incorporate your domain expertise.Starting Price: Free -
12
ZenML
ZenML
Simplify your MLOps pipelines. Manage, deploy, and scale on any infrastructure with ZenML. ZenML is completely free and open-source. See the magic with just two simple commands. Set up ZenML in a matter of minutes, and start with all the tools you already use. ZenML standard interfaces ensure that your tools work together seamlessly. Gradually scale up your MLOps stack by switching out components whenever your training or deployment requirements change. Keep up with the latest changes in the MLOps world and easily integrate any new developments. Define simple and clear ML workflows without wasting time on boilerplate tooling or infrastructure code. Write portable ML code and switch from experimentation to production in seconds. Manage all your favorite MLOps tools in one place with ZenML's plug-and-play integrations. Prevent vendor lock-in by writing extensible, tooling-agnostic, and infrastructure-agnostic code.Starting Price: Free -
13
Aporia
Aporia
Create customized monitors for your machine learning models with our magically-simple monitor builder, and get alerts for issues like concept drift, model performance degradation, bias and more. Aporia integrates seamlessly with any ML infrastructure. Whether it’s a FastAPI server on top of Kubernetes, an open-source deployment tool like MLFlow or a machine learning platform like AWS Sagemaker. Zoom into specific data segments to track model behavior. Identify unexpected bias, underperformance, drifting features and data integrity issues. When there are issues with your ML models in production, you want to have the right tools to get to the root cause as quickly as possible. Go beyond model monitoring with our investigation toolbox to take a deep dive into model performance, data segments, data stats or distribution. -
14
Amazon EC2 Trn1 Instances
Amazon
Amazon Elastic Compute Cloud (EC2) Trn1 instances, powered by AWS Trainium chips, are purpose-built for high-performance deep learning training of generative AI models, including large language models and latent diffusion models. Trn1 instances offer up to 50% cost-to-train savings over other comparable Amazon EC2 instances. You can use Trn1 instances to train 100B+ parameter DL and generative AI models across a broad set of applications, such as text summarization, code generation, question answering, image and video generation, recommendation, and fraud detection. The AWS Neuron SDK helps developers train models on AWS Trainium (and deploy models on the AWS Inferentia chips). It integrates natively with frameworks such as PyTorch and TensorFlow so that you can continue using your existing code and workflows to train models on Trn1 instances.Starting Price: $1.34 per hour -
15
Amazon EC2 Inf1 Instances
Amazon
Amazon EC2 Inf1 instances are purpose-built to deliver high-performance and cost-effective machine learning inference. They provide up to 2.3 times higher throughput and up to 70% lower cost per inference compared to other Amazon EC2 instances. Powered by up to 16 AWS Inferentia chips, ML inference accelerators designed by AWS, Inf1 instances also feature 2nd generation Intel Xeon Scalable processors and offer up to 100 Gbps networking bandwidth to support large-scale ML applications. These instances are ideal for deploying applications such as search engines, recommendation systems, computer vision, speech recognition, natural language processing, personalization, and fraud detection. Developers can deploy their ML models on Inf1 instances using the AWS Neuron SDK, which integrates with popular ML frameworks like TensorFlow, PyTorch, and Apache MXNet, allowing for seamless migration with minimal code changes.Starting Price: $0.228 per hour -
16
Amazon EC2 G5 Instances
Amazon
Amazon EC2 G5 instances are the latest generation of NVIDIA GPU-based instances that can be used for a wide range of graphics-intensive and machine-learning use cases. They deliver up to 3x better performance for graphics-intensive applications and machine learning inference and up to 3.3x higher performance for machine learning training compared to Amazon EC2 G4dn instances. Customers can use G5 instances for graphics-intensive applications such as remote workstations, video rendering, and gaming to produce high-fidelity graphics in real time. With G5 instances, machine learning customers get high-performance and cost-efficient infrastructure to train and deploy larger and more sophisticated models for natural language processing, computer vision, and recommender engine use cases. G5 instances deliver up to 3x higher graphics performance and up to 40% better price performance than G4dn instances. They have more ray tracing cores than any other GPU-based EC2 instance.Starting Price: $1.006 per hour -
17
Amazon Augmented AI (A2I)
Amazon
Amazon Augmented AI (Amazon A2I) makes it easy to build the workflows required for human review of ML predictions. Amazon A2I brings human review to all developers, removing the undifferentiated heavy lifting associated with building human review systems or managing large numbers of human reviewers. Many machine learning applications require humans to review low confidence predictions to ensure the results are correct. For example, extracting information from scanned mortgage application forms can require human review in some cases due to low-quality scans or poor handwriting. But building human review systems can be time consuming and expensive because it involves implementing complex processes or “workflows”, writing custom software to manage review tasks and results, and in many cases, managing large groups of reviewers. -
18
Privacera
Privacera
At the intersection of data governance, privacy, and security, Privacera’s unified data access governance platform maximizes the value of data by providing secure data access control and governance across hybrid- and multi-cloud environments. The hybrid platform centralizes access and natively enforces policies across multiple cloud services—AWS, Azure, Google Cloud, Databricks, Snowflake, Starburst and more—to democratize trusted data enterprise-wide without compromising compliance with regulations such as GDPR, CCPA, LGPD, or HIPAA. Trusted by Fortune 500 customers across finance, insurance, retail, healthcare, media, public and the federal sector, Privacera is the industry’s leading data access governance platform that delivers unmatched scalability, elasticity, and performance. Headquartered in Fremont, California, Privacera was founded in 2016 to manage cloud data privacy and security by the creators of Apache Ranger™ and Apache Atlas™. -
19
MLflow
MLflow
MLflow is an open source platform to manage the ML lifecycle, including experimentation, reproducibility, deployment, and a central model registry. MLflow currently offers four components. Record and query experiments: code, data, config, and results. Package data science code in a format to reproduce runs on any platform. Deploy machine learning models in diverse serving environments. Store, annotate, discover, and manage models in a central repository. The MLflow Tracking component is an API and UI for logging parameters, code versions, metrics, and output files when running your machine learning code and for later visualizing the results. MLflow Tracking lets you log and query experiments using Python, REST, R API, and Java API APIs. An MLflow Project is a format for packaging data science code in a reusable and reproducible way, based primarily on conventions. In addition, the Projects component includes an API and command-line tools for running projects. -
20
TruEra
TruEra
A machine learning monitoring solution that helps you easily oversee and troubleshoot high model volumes. With explainability accuracy that’s unparalleled and unique analyses that are not available anywhere else, data scientists avoid false alarms and dead ends, addressing critical problems quickly and effectively. Your machine learning models stay optimized, so that your business is optimized. TruEra’s solution is based on an explainability engine that, due to years of dedicated research and development, is significantly more accurate than current tools. TruEra’s enterprise-class AI explainability technology is without peer. The core diagnostic engine is based on six years of research at Carnegie Mellon University and dramatically outperforms competitors. The platform quickly performs sophisticated sensitivity analysis that enables data scientists, business users, and risk and compliance teams to understand exactly how and why a model makes predictions. -
21
Wallaroo.AI
Wallaroo.AI
Wallaroo facilitates the last-mile of your machine learning journey, getting ML into your production environment to impact the bottom line, with incredible speed and efficiency. Wallaroo is purpose-built from the ground up to be the easy way to deploy and manage ML in production, unlike Apache Spark, or heavy-weight containers. ML with up to 80% lower cost and easily scale to more data, more models, more complex models. Wallaroo is designed to enable data scientists to quickly and easily deploy their ML models against live data, whether to testing environments, staging, or prod. Wallaroo supports the largest set of machine learning training frameworks possible. You’re free to focus on developing and iterating on your models while letting the platform take care of deployment and inference at speed and scale. -
22
Galileo
Galileo
Models can be opaque in understanding what data they didn’t perform well on and why. Galileo provides a host of tools for ML teams to inspect and find ML data errors 10x faster. Galileo sifts through your unlabeled data to automatically identify error patterns and data gaps in your model. We get it - ML experimentation is messy. It needs a lot of data and model changes across many runs. Track and compare your runs in one place and quickly share reports with your team. Galileo has been built to integrate with your ML ecosystem. Send a fixed dataset to your data store to retrain, send mislabeled data to your labelers, share a collaborative report, and a lot more! Galileo is purpose-built for ML teams to build better quality models, faster. -
23
Fiddler AI
Fiddler AI
Fiddler is a pioneer in Model Performance Management for responsible AI. The Fiddler platform’s unified environment provides a common language, centralized controls, and actionable insights to operationalize ML/AI with trust. Model monitoring, explainable AI, analytics, and fairness capabilities address the unique challenges of building in-house stable and secure MLOps systems at scale. Unlike observability solutions, Fiddler integrates deep XAI and analytics to help you grow into advanced capabilities over time and build a framework for responsible AI practices. Fortune 500 organizations use Fiddler across training and production models to accelerate AI time-to-value and scale, build trusted AI solutions, and increase revenue. -
24
Amazon SageMaker Debugger
Amazon
Optimize ML models by capturing training metrics in real-time and sending alerts when anomalies are detected. Automatically stop training processes when the desired accuracy is achieved to reduce the time and cost of training ML models. Automatically profile and monitor system resource utilization and send alerts when resource bottlenecks are identified to continuously improve resource utilization. Amazon SageMaker Debugger can reduce troubleshooting during training from days to minutes by automatically detecting and alerting you to remediate common training errors such as gradient values becoming too large or too small. Alerts can be viewed in Amazon SageMaker Studio or configured through Amazon CloudWatch. Additionally, the SageMaker Debugger SDK enables you to automatically detect new classes of model-specific errors such as data sampling, hyperparameter values, and out-of-bound values. -
25
Amazon SageMaker Model Training reduces the time and cost to train and tune machine learning (ML) models at scale without the need to manage infrastructure. You can take advantage of the highest-performing ML compute infrastructure currently available, and SageMaker can automatically scale infrastructure up or down, from one to thousands of GPUs. Since you pay only for what you use, you can manage your training costs more effectively. To train deep learning models faster, SageMaker distributed training libraries can automatically split large models and training datasets across AWS GPU instances, or you can use third-party libraries, such as DeepSpeed, Horovod, or Megatron. Efficiently manage system resources with a wide choice of GPUs and CPUs including P4d.24xl instances, which are the fastest training instances currently available in the cloud. Specify the location of data, indicate the type of SageMaker instances, and get started with a single click.
-
26
Amazon SageMaker provides all the tools and libraries you need to build ML models, the process of iteratively trying different algorithms and evaluating their accuracy to find the best one for your use case. In Amazon SageMaker you can pick different algorithms, including over 15 that are built-in and optimized for SageMaker, and use over 150 pre-built models from popular model zoos available with a few clicks. SageMaker also offers a variety of model-building tools including Amazon SageMaker Studio Notebooks and RStudio where you can run ML models on a small scale to see results and view reports on their performance so you can come up with high-quality working prototypes. Amazon SageMaker Studio Notebooks help you build ML models faster and collaborate with your team. Amazon SageMaker Studio notebooks provide one-click Jupyter notebooks that you can start working within seconds. Amazon SageMaker also enables one-click sharing of notebooks.
-
27
Amazon SageMaker Studio
Amazon
Amazon SageMaker Studio is an integrated development environment (IDE) that provides a single web-based visual interface where you can access purpose-built tools to perform all machine learning (ML) development steps, from preparing data to building, training, and deploying your ML models, improving data science team productivity by up to 10x. You can quickly upload data, create new notebooks, train and tune models, move back and forth between steps to adjust experiments, collaborate seamlessly within your organization, and deploy models to production without leaving SageMaker Studio. Perform all ML development steps, from preparing raw data to deploying and monitoring ML models, with access to the most comprehensive set of tools in a single web-based visual interface. Amazon SageMaker Unified Studio is a comprehensive, AI and data development environment designed to streamline workflows and simplify the process of building and deploying machine learning models. -
28
Amazon SageMaker Studio Lab
Amazon
Amazon SageMaker Studio Lab is a free machine learning (ML) development environment that provides the compute, storage (up to 15GB), and security, all at no cost, for anyone to learn and experiment with ML. All you need to get started is a valid email address, you don’t need to configure infrastructure or manage identity and access or even sign up for an AWS account. SageMaker Studio Lab accelerates model building through GitHub integration, and it comes preconfigured with the most popular ML tools, frameworks, and libraries to get you started immediately. SageMaker Studio Lab automatically saves your work so you don’t need to restart in between sessions. It’s as easy as closing your laptop and coming back later. Free machine learning development environment that provides the computing, storage, and security to learn and experiment with ML. GitHub integration and preconfigured with the most popular ML tools, frameworks, and libraries so you can get started immediately. -
29
Amazon SageMaker Feature Store is a fully managed, purpose-built repository to store, share, and manage features for machine learning (ML) models. Features are inputs to ML models used during training and inference. For example, in an application that recommends a music playlist, features could include song ratings, listening duration, and listener demographics. Features are used repeatedly by multiple teams and feature quality is critical to ensure a highly accurate model. Also, when features used to train models offline in batch are made available for real-time inference, it’s hard to keep the two feature stores synchronized. SageMaker Feature Store provides a secured and unified store for feature use across the ML lifecycle. Store, share, and manage ML model features for training and inference to promote feature reuse across ML applications. Ingest features from any data source including streaming and batch such as application logs, service logs, clickstreams, sensors, etc.
-
30
Amazon SageMaker Data Wrangler reduces the time it takes to aggregate and prepare data for machine learning (ML) from weeks to minutes. With SageMaker Data Wrangler, you can simplify the process of data preparation and feature engineering, and complete each step of the data preparation workflow (including data selection, cleansing, exploration, visualization, and processing at scale) from a single visual interface. You can use SQL to select the data you want from a wide variety of data sources and import it quickly. Next, you can use the Data Quality and Insights report to automatically verify data quality and detect anomalies, such as duplicate rows and target leakage. SageMaker Data Wrangler contains over 300 built-in data transformations so you can quickly transform data without writing any code. Once you have completed your data preparation workflow, you can scale it to your full datasets using SageMaker data processing jobs; train, tune, and deploy models.
