For each instance, there is set of atomic numbers (input_ids), 3-D coordinates (coords), a set of forces per atom (forces), the total and formation energy per
system… See the full description on the dataset page: https://huggingface.co/datasets/vector-institute/s2ef-15m.