ai4science-studio/models.yaml at main · AMDResearch/ai4science-studio · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
# AI4Science Studio — Model Index
#
# Machine-readable index of all models in this repository.
# Agents: read this file first to discover what's available.
# For full details on any model, read <domain>/models/<slug>/model.yaml.

models:
  # --- Earth Science ---
  - slug: StormCast
    domain: earth_science
    hf_id: nvidia/stormcast-v1-era5-hrrr
    license: Apache-2.0
    task: High-resolution convection-allowing weather prediction
    tasks_available: [inference, ensemble]
    path: earth_science/models/StormCast

  - slug: ORBIT-2
    domain: earth_science
    hf_id: jychoi-hpc/ORBIT-2
    license: See upstream model card
    task: Global weather and climate downscaling
    tasks_available: [inference]
    path: earth_science/models/ORBIT-2

  - slug: ArchesWeather
    domain: earth_science
    hf_id: gcouairon/ArchesWeather
    license: See upstream repository and Hugging Face model card
    task: Global AI weather forecasting at 1.5° resolution
    tasks_available: [inference, train]
    path: earth_science/models/ArchesWeather

  - slug: Aurora
    domain: earth_science
    hf_id: microsoft/aurora
    license: CC-BY-NC-SA-4.0
    task: Foundation model for Earth system forecasting at 0.1° resolution
    tasks_available: [inference]
    path: earth_science/models/Aurora

  - slug: GenCast
    domain: earth_science
    hf_id: N/A
    license: CC-BY-NC-SA-4.0
    task: Diffusion-based probabilistic ensemble weather forecasting
    tasks_available: [inference]
    path: earth_science/models/GenCast

  - slug: NeuralGCM
    domain: earth_science
    hf_id: N/A
    license: "Code: Apache-2.0 · Weights: CC-BY-SA-4.0"
    task: Hybrid physics + ML atmospheric model
    tasks_available: [inference]
    path: earth_science/models/NeuralGCM

  - slug: PanguWeather
    domain: earth_science
    hf_id: N/A
    license: CC-BY-NC-SA-4.0
    task: Transformer-based deterministic global weather forecasting
    tasks_available: [inference]
    path: earth_science/models/PanguWeather

  # --- Material Science ---
  - slug: MatterGen
    domain: material_science
    hf_id: microsoft/mattergen
    license: MIT
    task: Diffusion-based generative model for inorganic materials design
    tasks_available: [inference, train]
    path: material_science/models/MatterGen

  - slug: HydraGNN
    domain: material_science
    hf_id: mlupopa/HydraGNN_Predictive_GFM_2024
    license: BSD-3-Clause-Clear
    task: Ensemble of predictive graph foundation models for atomistic materials
    tasks_available: [inference, train]
    path: material_science/models/HydraGNN

  # --- Healthcare & Life Sciences ---
  - slug: GP-MoLFormer
    domain: healthcare
    hf_id: ibm-research/GP-MoLFormer-Uniq
    license: Apache-2.0
    task: SMILES molecule generation
    tasks_available: [inference, finetune]
    path: healthcare/models/GP-MoLFormer

  - slug: SwinUNETR
    domain: healthcare
    hf_id: N/A
    license: Apache-2.0
    task: 3D medical image segmentation
    tasks_available: [train, inference]
    path: healthcare/models/SwinUNETR

  - slug: SemlaFlow
    domain: healthcare
    hf_id: N/A
    license: MIT
    task: 3D molecular structure generation
    tasks_available: [inference]
    path: healthcare/models/SemlaFlow

  - slug: REINVENT4
    domain: healthcare
    hf_id: N/A
    license: Apache-2.0
    task: Reinforcement learning for molecular design
    tasks_available: [finetune]
    path: healthcare/models/REINVENT4

  # --- Physics Simulation ---
  - slug: MATEY
    domain: physics_simulation
    hf_id: N/A
    license: MIT
    task: Spatiotemporal surrogate modeling for multiphysics systems
    tasks_available: [train, inference]
    path: physics_simulation/models/MATEY

  - slug: Walrus
    domain: physics_simulation
    hf_id: polymathic-ai/walrus
    license: MIT
    task: Cross-domain surrogate simulation for continuum dynamics
    tasks_available: [inference]
    path: physics_simulation/models/Walrus