在代碼的世界里追尋創(chuàng)世之樂

●個個尹

胡淵鳴——一個非凡的普通科研人。

2010年高一即獲得清華大學(xué)保送資格，2013年正式進(jìn)入清華大學(xué)姚班。本科期間，先后前往東京大學(xué)、斯坦福大學(xué)訪學(xué)。2017年，進(jìn)入麻省理工學(xué)院讀博。入學(xué)13個月后，拿到麻省理工學(xué)院碩士學(xué)位。

本科期間在微軟亞洲研究院實習(xí)，這是微軟公司在美國本土以外最大的基礎(chǔ)研究機構(gòu)。從事深度學(xué)習(xí)和計算機圖形學(xué)研究，本科便有多篇論文中選CVPR（美國電氣和電子工程師協(xié)會國際計算機視覺和模式識別會議）、SIGGRAPH（計算機圖形學(xué)會議）等國際頂會。

事實上，上述短短幾行履歷遠(yuǎn)遠(yuǎn)不能概括出胡淵鳴。最近，讓胡淵鳴為大家熟知的是他的一篇文章——《99行代碼的“冰雪奇緣”》。99行代碼雖短，背后的技術(shù)故事卻很長。故事要從物質(zhì)點法說起，這是一種在影視特效領(lǐng)域廣受青睞的模擬連續(xù)介質(zhì)方法，《冰雪奇緣》就用到了它。早期，它運行速度非常慢，比如《冰雪奇緣》里安娜過雪地的鏡頭，據(jù)說要在集群上跑整整一個星期。大四畢業(yè)的暑假，胡淵鳴投入了移動最小二乘物質(zhì)點法的研究，為的就是提高物質(zhì)點法的運行速度和性能。

胡淵鳴的靈感是，用移動最小二乘法統(tǒng)一仿射粒子細(xì)胞法中的仿射梯度場和物質(zhì)點法中的變形梯度更新兩種離散化。在賓夕法尼亞大學(xué)蔣陳凡夫教授的指導(dǎo)下，胡淵鳴等人完成了移動最小二乘物質(zhì)點法方法的研究，不僅實現(xiàn)了新的應(yīng)力散度離散化，使物質(zhì)點法的運行速度快了兩倍，還成功模擬了物質(zhì)點法此前并不支持的各種新現(xiàn)象。這項成果最終發(fā)表在SIGGRAPHP。

值得一提的是，雖然很多報道把重點放在了99行代碼上，但胡淵鳴真正令人崇拜的地方不僅于此。代碼行數(shù)只是表象，胡淵鳴令人稱贊的是創(chuàng)造了Taichi（太極）這個編程語言。相比TensorFlow（開放源代碼軟件庫，用于進(jìn)行高性能數(shù)值計算）和PyTorch（開源的Python機器學(xué)習(xí)庫，用于自然語言處理等應(yīng)用程序），太極的渲染速度分別比前者快了188倍和13.4倍，并且能夠模擬雪等物理流體，創(chuàng)造出類似《冰雪奇緣》中的魔法場景的場景。

毫無疑問，太極編程語言的工作量非常大。這是一個全新的系統(tǒng)，項目早期要做的設(shè)計決策非常多，需要花費大量的時間和精力。有一次，胡淵鳴連續(xù)在家寫代碼，最后，他實在受不了沒人說話的生活，就去實驗室找同學(xué)聊天，結(jié)果被同學(xué)調(diào)侃“身上‘長蘑菇’了”。

太極的編寫過程不容易。一開始，胡淵鳴低估了寫編譯器的工作量，沒有完成一個月寫出一個理想編譯器的設(shè)想。其中，因為第一次寫的緣故，他缺乏設(shè)計匯編者的經(jīng)驗，還把IR（中間表現(xiàn)形式）設(shè)計錯了，最后只好砍掉重來……正是他不厭其煩的改進(jìn)和堅持，才有了現(xiàn)在Hierarchical SSA（一種分層靜態(tài)單賦值）的結(jié)構(gòu)，使得很多優(yōu)化成為可能，促成了太極編程的誕生。

用太極語言編寫的模擬器大大簡化了代碼，可微分彈性對象模擬器只用了110行代碼，而直接用CUDA（由NVIDIA推出的通用并行計算架構(gòu)，該架構(gòu)使GPU能夠解決復(fù)雜的計算問題）編寫則需要490行。同時，太極的速度還很快，相比CUDA版本幾乎沒有什么損失。而且，神經(jīng)網(wǎng)絡(luò)控制器一般只需要幾十次迭代，即可完成優(yōu)化。

近年來，Python這門計算機語言因為入門方便，實用性高，熱度極速上升，不少人紛紛通過它開啟編程的大門。考慮到Python的這一特性，胡淵鳴轉(zhuǎn)變了原來要將太極做成一門單獨的編程語言的方向，而是把太極改成了pypi（Python軟件包索引）安裝包，讓太極能夠基于Python運行，他還為此編寫了完整的使用教程。這些做法既體現(xiàn)了胡淵鳴縝密的思維，又進(jìn)一步地擴大了太極的使用范圍，降低了使用難度。

網(wǎng)友看了胡淵鳴的事跡后，不由感嘆道：“圖形+系統(tǒng)+編譯，真是創(chuàng)世的快樂?！边@個評價十分貼切，他也曾表達(dá)了自己在鉆研編譯器中獲得的快樂。每天對于太極編程語言相關(guān)問題的思考，大大豐富了他自己的精神世界，把人世間的種種噪聲都隔絕了，這讓他的生活質(zhì)量有了很大的提高。

這個時代缺少學(xué)霸或科研人員嗎？或許并不是，比起學(xué)霸光環(huán)，更難能可貴的是，在自己擅長的領(lǐng)域，享受鉆研的樂趣。飽含一腔熱情并愿意靜下心來做真正有意義的事情，這正是胡淵鳴最讓人佩服的地方。

（王世全摘自《知識窗》2021年第1期）