mlip_arena/models/registry.yaml

MACE_MP_Medium:
  module: mace
  username: cyrusyc # HF username
  datetime: 2024-03-25T14:30:00 # TODO: Fake datetime
  datasets: # list of training datasets
    - atomind/mptrj # TODO: fake HF dataset repo
  cpu-tasks:
    - alexandria
    - qmof
  gpu-tasks:
    - diatomics

# CHGNet:
#   module: chgnet
#   username: cyrusyc
#   datetime: 2024-03-25T14:30:00
#   datasets:
#     - atomind/mptrj
#   cpu-tasks:
#     - diatomics