今天是 OneFlow 开源的第 903 天,OneFlow v0.9.0 正式发布。本次更新包含 640 个 commit,完整更新列表请查看链接:https://github.com/Oneflow-Inc/oneflow/releases/tag/v0.9.0,欢迎下载体验新版本,期待你的反馈。
OneFlow v0.9.0 主要包括以下新增亮点功能和优化:
1. PyTorch 兼容性
OneFlow 对 PyTorch API 和 模型的兼容性进一步提升。新增与 PyTorch 对齐的 86 个 API 接口 和 算子, 修复了 104 个算子兼容性相关的 bug。支持更多的 PyTorch 模型一键迁移成 OneFlow 模型,并获得性能加速。
支持 Stable Diffusion、GLM、YOLOv5 等模型一键迁移为 OneFlow 模型
oneflow.load 支持直接加载 torch.save 的模型,模型迁移更方便