StableHLO:跨框架机器学习操作的开源解决方案

【免费下载链接】stablehlo Backward compatible ML compute opset inspired by HLO/MHLO 【免费下载链接】stablehlo 项目地址: https://gitcode.com/gh_mirrors/st/stablehlo

StableHLO 是一个开源项目,旨在为机器学习模型提供高级操作的便携性层。它通过创建不同机器学习框架(如 TensorFlow、JAX 和 PyTorch)和机器学习编译器(如 XLA 和 IREE)之间的互操作性,简化并加速机器学习开发。

1. 项目基础介绍及主要编程语言

StableHLO 项目托管在 GitHub 上,使用 MLIR(Multi-Level Intermediate Representation)作为其核心构建块,主要的编程语言包括 MLIR、C++ 和 Python。MLIR 是一种灵活的中间表示,它允许在不同的编译器和工具之间共享和转换代码。

2. 项目的核心功能

StableHLO 的核心功能是提供一个向后兼容的机器学习计算操作集,它基于 MHLO(Multi-Level High-Level Operations)方言,并增强了其功能,包括序列化和版本控制。以下是该项目的几个关键特性:

  • 跨框架兼容性:StableHLO 为不同框架之间提供了一种通用语言,使得一个框架生成的模型可以被另一个框架的编译器使用。
  • 序列化和版本控制:通过使用 MLIR 字节码作为序列化格式,StableHLO 提供了前后兼容性保证,确保框架和编译器之间的兼容性,即使 StableHLO 继续发展。
  • 灵活的构建和部署:项目提供了详细的构建指南,支持在 Linux 和 macOS 上构建,并可以生成 Python 绑定,方便在多种环境中使用。

3. 项目最近更新的功能

最近的更新主要关注于性能提升、错误修复和功能的增强。以下是一些最近的更新内容:

  • 性能优化:通过改进构建过程和引入新的编译器优化,提高了 StableHLO 的性能。
  • 错误修复:修复了在编译和运行时遇到的各种问题,提高了稳定性和可靠性。
  • 功能增强:增加了对 TensorFlow SavedModel 的转换工具,允许用户将 StableHLO 程序转换为 TensorFlow SavedModel,从而更容易在 TensorFlow 生态系统中使用。

StableHLO 项目的持续发展受益于开源社区的贡献,为机器学习开发者提供了强大的工具,以实现跨框架的模型开发和部署。

【免费下载链接】stablehlo Backward compatible ML compute opset inspired by HLO/MHLO 【免费下载链接】stablehlo 项目地址: https://gitcode.com/gh_mirrors/st/stablehlo

Logo

脑启社区是一个专注类脑智能领域的开发者社区。欢迎加入社区,共建类脑智能生态。社区为开发者提供了丰富的开源类脑工具软件、类脑算法模型及数据集、类脑知识库、类脑技术培训课程以及类脑应用案例等资源。

更多推荐