Add functions to profile model performance #282
Description
It would be useful to have the following profiling tools for a given potential:
- compute graph with allocated memory blocks
- force calculation timing/GPU memory consumption as a function of atoms in a periodic box
- GPU memory trace for force calculations