xnn.gnn.models.aimnet2_foundation.FOUNDATION_MODELS

xnn.gnn.models.aimnet2_foundation.FOUNDATION_MODELS = {'aimnet2-b973c-2025-d3-0': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_2025_b973c_d3_0.pt', 'MIT'), 'aimnet2-b973c-2025-d3-1': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_2025_b973c_d3_1.pt', 'MIT'), 'aimnet2-b973c-2025-d3-2': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_2025_b973c_d3_2.pt', 'MIT'), 'aimnet2-b973c-2025-d3-3': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_2025_b973c_d3_3.pt', 'MIT'), 'aimnet2-b973c-d3-0': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_b973c_d3_0.pt', 'MIT'), 'aimnet2-b973c-d3-1': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_b973c_d3_1.pt', 'MIT'), 'aimnet2-b973c-d3-2': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_b973c_d3_2.pt', 'MIT'), 'aimnet2-b973c-d3-3': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_b973c_d3_3.pt', 'MIT'), 'aimnet2-nse-0': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2NSE/aimnet2nse_wb97m_0.pt', 'MIT'), 'aimnet2-nse-1': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2NSE/aimnet2nse_wb97m_1.pt', 'MIT'), 'aimnet2-nse-2': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2NSE/aimnet2nse_wb97m_2.pt', 'MIT'), 'aimnet2-nse-3': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2NSE/aimnet2nse_wb97m_3.pt', 'MIT'), 'aimnet2-pd-0': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2Pd/aimnet2-pd_0.pt', 'MIT'), 'aimnet2-pd-1': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2Pd/aimnet2-pd_1.pt', 'MIT'), 'aimnet2-pd-2': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2Pd/aimnet2-pd_2.pt', 'MIT'), 'aimnet2-pd-3': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2Pd/aimnet2-pd_3.pt', 'MIT'), 'aimnet2-rxn-0': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2rxn/aimnet2_rxn_0.pt', 'MIT'), 'aimnet2-rxn-1': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2rxn/aimnet2_rxn_1.pt', 'MIT'), 'aimnet2-rxn-2': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2rxn/aimnet2_rxn_2.pt', 'MIT'), 'aimnet2-rxn-3': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2rxn/aimnet2_rxn_3.pt', 'MIT'), 'aimnet2-wb97m-d3-0': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_wb97m_d3_0.pt', 'MIT'), 'aimnet2-wb97m-d3-1': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_wb97m_d3_1.pt', 'MIT'), 'aimnet2-wb97m-d3-2': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_wb97m_d3_2.pt', 'MIT'), 'aimnet2-wb97m-d3-3': ('https://storage.googleapis.com/aimnetcentral/aimnet2v2/AIMNet2/aimnet2_wb97m_d3_3.pt', 'MIT')}

name -> (download URL, license), a view of the “aimnet2” entries of the model hub registry (models.json), which is the single list. Every family is MIT-licensed.

Type:

The published checkpoints