模块化:MAX 24.3 - 介绍MAX引擎可扩展性

模块化:MAX 24.3 - 介绍MAX引擎可扩展性

💡 原文英文,约1200词,阅读约需5分钟。
📝

内容提要

MAX 24.3发布,包括MAX Engine Extensibility API预览,允许开发者在新一代编译器和运行时堆栈上统一、编程和组合AI流水线,以实现最佳性能。MAX Engine是一个用于运行AI推理的新一代编译器和运行时库,支持PyTorch(TorchScript)、ONNX和本地Mojo模型,可在各种硬件上实现低延迟、高吞吐量的推理。此外,MAX平台通过使用MAX Graph API创建定制推理模型,使您能够充分发挥MAX Engine的潜力。MAX 24.3的新功能包括自定义操作,使构建AI流水线时更容易使用自定义操作。通过MAX Graphs和Mojo,您可以编写自己的MAX Graphs,并使用自定义操作扩展。MAX Engine和Mojo建立在MLIR之上,充分利用现代编译器的优势。MAX Engine确保您的Mojo代码在各种平台和硬件设置上可移植,扩大了部署选项。

🔎

延伸解读

自定义操作:解决传统框架的痛点

在PyTorch或ONNX中实现自定义操作通常面临性能低下、需要底层语言专业知识、跨硬件兼容性差以及维护困难等问题。MAX 24.3通过Mojo语言和统一的API简化了这一过程,开发者可以轻松编写、打包和部署自定义操作,无需深入C++或CUDA,同时保证跨平台一致性。

MAX Graph API:统一AI工作流

MAX Graph API允许开发者使用Mojo编写自定义操作并构建完整的推理图,实现AI工作流的集中管理。通过单一注解,内核可在不同机器学习框架中重用,减少不一致性。此外,MAX Engine提供自动内核融合和图优化,无需传统构建流程,加速开发迭代。

基于MLIR的零成本抽象与可移植性

MAX Engine和Mojo构建在MLIR之上,充分利用现代编译器优化。代码被内联并直接参与编译器的优化阶段,提升效率并减少开销。同时,只要平台支持LLVM或MLIR,Mojo代码即可移植,为开发者提供硬件选择自由,扩大部署选项。

未来路线图:MacOS、量化与GPU支持

MAX 24.3是MAX路线图的起点,即将推出MacOS支持和量化功能,GPU支持也计划于今年夏天发布。这些更新将进一步扩展MAX Engine的适用场景,为开发者提供更多部署灵活性和性能优化选项。

❓

Q&A

MAX 24.3的主要新功能是什么?

MAX 24.3的主要新功能包括MAX Engine Extensibility API预览和自定义操作,简化了AI流水线的构建过程。

MAX Engine如何支持AI推理?

MAX Engine是一个新一代编译器和运行时库,支持PyTorch、ONNX和本地Mojo模型,提供低延迟和高吞吐量的推理。

自定义操作在MAX 24.3中有什么优势?

自定义操作允许开发者定义特定任务的操作,简化了在AI流水线中实现复杂算法的过程。

如何使用Mojo编写自定义操作?

可以通过编写Mojo代码并使用@register.op装饰器来定义自定义操作,然后将其打包并加载到MAX Engine中。

MAX Graph API的作用是什么?

MAX Graph API集中化AI工作流,允许开发者重用内核,减少不一致性,提高开发效率。

MAX 24.3的未来更新计划有哪些?

未来更新计划包括对MacOS的支持和量化功能,预计将在夏季推出GPU支持。

🏷️

标签

➡️

继续阅读