Colossal-AI
ChatGPT低成本復現(xiàn)流程開源!任意單張消費級顯卡可體驗,顯存需求低至1.62GB
單機訓練速度提升7.73倍
比AlphaFold2快10倍!國產(chǎn)開源項目加速蛋白質(zhì)預測,超長序列22分鐘就搞定
全球最快的復合物結構預測模型
霸榜GitHub熱門第一多日后,Colossal-AI正式版發(fā)布
本次正式版更新重點優(yōu)化了分布式訓練性能及開發(fā)者的易用性,主要亮點包括:重構ZeRO以改善性能和易用性;添加細粒度Profiler TensorBoard監(jiān)控插件,監(jiān)測訓練過程中內(nèi)存、網(wǎng)絡等狀態(tài);更靈活的checkpoint策略,可擴展的pipeline模塊;開源蛋白質(zhì)預測FastFold等豐富行業(yè)解決方案