Skip to content

HFAiLab/hfai-models

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

幻方 AI 模型仓库

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

依赖

模型列表

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

Releases

No releases published

Packages

No packages published

Languages