use built-in Python tools: https://docs.python.org/3/library/profile.html

update: we may need to reorganize the driver(s) in order for this to work with MPI scheduler