DeepMind 之前曾在 GitHub 开源过一款 AI 程序:AlphaFold,其主要目的,是辅助人类完成生命探索的一大工程:蛋白质结构预测。
因其在该领域做出的杰出贡献,AlphaFold 被 Science 和 Nature 评选为「2021 年十大科学突破之首」。
但是,这款 AI 程序也存在一个问题,就是模型训练时间比较长。
有鉴于此,国内一家名为潞晨科技的 AI 公司,在 GitHub 上开源了一套训练推理加速方案:FastFold。
通过引入 GPU 优化和大模型训练技术,可将 AlphaFold 总体训练时间 从 11 天减少到 67 小时,且总成本更低,在长序列推理中也实现 9.3 ∼ 11.6 倍提升,加速效果颇为显著。
GitHub:github.com/hpcaitech/FastFold
目前,该项目的实现代码与算法,已开源至 GitHub。