旷视正式宣布开源深度学习框架“天元” 降低AI开发门槛

【小熊在线网 新闻报道】3月25日，旷视科技宣布开源新一代AI生产力平台Brain++的核心深度学习框架旷视天元（MegEngine）。天元框架通过训练、推理一体化的机制省去模型转换的过程，可直接将训练后的模型进行推理，并保证跨设备的模型精度对齐，同时天元内置自动模型优化、简化流程，减少因手工操作的机会，降低出错概率。此外，天元配置了Python和C++接口，支持动态图、静态图一键转换和混合编程，可使用高级编程语言进行图优化和图编译。在运行时管理环节，天元具备执行流与调度器，采用动态、静态内存分配并存的方式，并通过自动的亚线性内存管理优化器可以得到更好的内存优化效果；在底层设计上，天元的计算内核兼容主流计算设备，且支持多机多卡和分布式训练。而为了解决模型复现困难的问题，天元则支持 PyTorch Module 导入，可针对计算机视觉任务进行优化。