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