binzhu-ece / hfai-models

HFAI deep learning models

幻方 AI 模型仓库

这里是幻方 AI 的模型仓库，包含了从计算机视觉、自然语言处理到生物计算、气象预测等各个领域方面的模型，这些模型结合了幻方萤火超算集群的特点，使用并行训练、高效算子、高性能存储等多种方式，大幅提升原有模型的性能，节省训练时间。

依赖

• torch >= 1.8
• hfai
• ffrecord

模型列表

Tags:

• hfai.nn.to_hfai: 是否使用了算子加速
• hfai DDP: 是否使用了 HFAI DDP 加速
• hfai.datasets: 是否使用了 HFAI 数据仓库
• haiscale: 是否使用了 haiscale 并行训练库

1. 生物计算

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   AlphaFold2	@mingchuan	Yes	Yes	No

2. 自动驾驶

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   HDMapNet	@bixiao	No	No	Yes
   BEVFormer	@bixiao	Yes	No	Yes

3. 时间序列

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   LTSF-formers	@wenjie	No	No	Yes

4. 多模态

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   CLIP	@chengqi	Yes	Yes	Yes
   CLIP-GEN	@chengqi	No	Yes	Yes

5. 计算机视觉

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   ResNet50	@hfai	No	No	Yes
   MaskedAutoEncoder	@chengqi	Yes	Yes	Yes

6. 自然语言处理

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   BERT	@hfai	Yes	Yes	Yes

   name	maintainer	haiscale
   GPT2	@chengqi	Yes

7. 气象预测

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   OpenCastKit	@wenjie	Yes	Yes	Yes

8. 图神经网络

   name	maintainer	hfai.nn.to_hfai	hfai DDP	hfai.datasets
   DeepGCNs	@wenjie	Yes	Yes	Yes