Software Engineering

New Features: Improved Sagemaker Orchestration, New DAG Visualizer, Skypilot with Kubernetes, and more

ZenML Team
Oct 1, 2024
3 min

We are thrilled the announce the next release of ZenML, 0.67.0. Our recent update introduces several new features and enhancements. Here's an overview of the key changes:

Enhanced SageMaker Orchestration

Our latest release brings significant improvements to the SageMaker Orchestrator. It now supports the usage of TrainingJobs, that can use warm pools for AWS SageMaker to enhance overall performance and reduce startup times. This upgrade not only improves efficiency but may also lead to cost reductions by maintaining a pool of pre-initialized instances.

Revamped DAG Visualizer

The DAG Visualizer for Pipeline Runs has undergone a major overhaul. Users can now preview the actual DAG before pipeline completion, providing early insights into workflow structure. We've also made visual adjustments to improve clarity, making complex pipelines easier to understand.

Flexible Environment Configuration

We've introduced a powerful new feature for configuration flexibility. Users can now reference environment variables in both code and configuration files using the syntax ${ENV_VARIABLE_NAME}. This enhancement significantly improves adaptability across different environments and deployments.

Improved Cloud Integration Experience

For users working with major cloud platforms (AWS, Azure, and GCP), we've enhanced the experience by displaying direct pipeline and log URLs. This improvement streamlines access to critical information, saving time and reducing friction in workflow management.

Kubernetes Support for Skypilot

In a move to expand orchestration options, we've added compatibility for running the Skypilot orchestrator on Kubernetes clusters. This addition allows users to leverage Kubernetes for resource management and scaling of Skypilot workloads, offering greater flexibility in infrastructure choices.

Updated Deepchecks Integration

Our integration with Deepchecks has been refreshed to incorporate the latest features and improvements from the Deepchecks platform. This update enhances data validation and testing capabilities within workflows, providing users with more robust tools for ensuring data quality.

Expanded Documentation and Bug Fixes

We've extensively updated our documentation, covering integrations such as Lightning AI, Kubeflow, Comet, Neptune, Hugging Face deployer, and Weights & Biases. New documentation for run templates has also been added and numerous bug fixes have been implemented.

Conclusion

From major enhancements to core components like the SageMaker Orchestrator and DAG Visualizer, to the introduction of flexible environment variable references and expanded Kubernetes support, these updates aim to provide a more powerful, and user-friendly experience.

We encourage you to update to ZenML 0.67.0 and explore these new features. As always, we welcome your feedback and look forward to seeing how these improvements accelerate your ML projects.

For a complete list of changes and improvements, please check out our full release notes on GitHub.

Happy coding with ZenML!

Looking to Get Ahead in MLOps & LLMOps?

Subscribe to the ZenML newsletter and receive regular product updates, tutorials, examples, and more articles like this one.
We care about your data in our privacy policy.