Skip to content

Commit fd865cc

Browse files
authored
Update readme plot (#236)
1 parent 71e1d41 commit fd865cc

File tree

2 files changed

+185
-5
lines changed

2 files changed

+185
-5
lines changed

README.md

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -92,14 +92,15 @@ print(relaxed_state.energy)
9292

9393
TorchSim achieves up to 100x speedup compared to ASE with popular MLIPs.
9494

95-
![Speedup comparison](/docs/_static/speedup_plot.svg)
95+
<img src="/docs/_static/speedup_plot.svg" alt="Speedup comparison" width="100%">
9696

9797
This figure compares the time per atom of ASE and `torch_sim`. Time per atom is defined
9898
as the number of atoms / total time. While ASE can only run a single system of `n_atoms`
9999
(on the $x$ axis), `torch_sim` can run as many systems as will fit in memory. On an H100 80 GB card,
100-
the max atoms that could fit in memory was ~8,000 for [GemNet](https://github.com/FAIR-Chem/fairchem), ~10,000 for [MACE](https://github.com/ACEsuit/mace), and ~2,500
101-
for [SevenNet](https://github.com/MDIL-SNU/SevenNet). This metric describes model performance by capturing speed and memory
102-
usage simultaneously.
100+
the max atoms that could fit in memory was ~8,000 for [EGIP](https://github.com/FAIR-Chem/fairchem),
101+
~10,000 for [MACE-MPA-0](https://github.com/ACEsuit/mace), ~22,000 for [Mattersim V1 1M](https://github.com/microsoft/mattersim),
102+
~2,500 for [SevenNet](https://github.com/MDIL-SNU/SevenNet), and ~9000 for [PET-MAD](https://github.com/lab-cosmo/pet-mad).
103+
This metric describes model performance by capturing speed and memory usage simultaneously.
103104

104105
## Installation
105106

0 commit comments

Comments
 (0)