See https://unyt.readthedocs.io/en/stable/ in Python. This sort of feature fits really well within Python's featureset.