Stats for activations (data top_vecs)
Stats for params (update diff and data)
Stats for grtadients (diff)