MegEngine（天元）是一個快速，可擴(kuò)展，易于使用且支持自動求導(dǎo)的深度學(xué)習(xí)框架，中文名為“天元”，是曠視 AI 戰(zhàn)略的重要組成部分，負(fù)責(zé) AI 三要素（算法，算力，數(shù)據(jù)）中的“算法”。

特點：

一是“訓(xùn)練推理一體”。MegEngine 支持多種硬件平臺（ CPU，GPU，ARM ）。不同硬件上的推理框架和 MegEngine 的訓(xùn)練框架無縫銜接。部署時無需做額外的模型轉(zhuǎn)換，速度/精度和訓(xùn)練保持一致，有效解決了 AI 落地中“部署環(huán)境和訓(xùn)練環(huán)境不同，部署難”的問題。

二是“動靜合一”。動態(tài)圖易調(diào)試，靜態(tài)圖好部署。魚和熊掌如何兼得，是現(xiàn)代深度學(xué)習(xí)框架的核心訴求。MegEngine 在靜態(tài)圖的基礎(chǔ)上，逐漸加入支持完整動態(tài)圖的功能。在動態(tài)模式下加速研發(fā)過程，無需改變模型代碼一鍵切換至靜態(tài)模式下的部署，為科研和算法工程師同時提供便利。

三是“兼容并包”。MegEngine 的頂層 API 基于 Python，采取了類似于 PyTorch 的風(fēng)格。簡單直接，易于上手，便于現(xiàn)有項目進(jìn)行移植或整合。為更好地幫助學(xué)習(xí)實踐，MegEngine 同時提供了“開箱即用”的在線深度學(xué)習(xí)工具?MegStudio?，和匯聚了頂尖算法和模型的預(yù)訓(xùn)練模型集合?Model Hub?。

四是“靈活高效”。MegEngine 底層的高性能算子庫對于不同的硬件架構(gòu)進(jìn)行了深度適配和優(yōu)化，并提供高效的亞線性內(nèi)存優(yōu)化策略，對于生產(chǎn)環(huán)境繁多的計算設(shè)備提供了極致的性能保證。高效易用的分布式訓(xùn)練實現(xiàn)能有效支持富有彈性的大規(guī)模訓(xùn)練。