[MPI] Add tokens to the statistics

Created by: inailuig

Adds (optional) tokens to all relevant functions in nk.stats.

Let me know if I should change the last two commits to not return the token to avoid breaking compatibility of possible external users of nk.stats.statistics for now