Moved from docs ## TODO - [ ] Remove atom_modify / compute virial dependency. - [ ] Add support for ASE as calculator interface. - [ ] Add support for multi GPUs (with `e3gnn/parallel`). - [ ] Implement without Unified Memory. - [ ] Unfix the `threadsPerBlock=128`. - [ ] Unroll the repetition loop `k` (for small number of atoms).