Use the numpy equivalents on the Quantity. See https://github.com/ai2cm/pace/pull/375#discussion_r1019721303.