Refactor memory use function #474
Description
Refactoring the mem use function so that it does not allocate memory and is decoupled from the trajectory array.
{{ message }}
Refactoring the mem use function so that it does not allocate memory and is decoupled from the trajectory array.