As far as I recall Meta's Galactica has special tokens to mark working out which can include python code and can run the code during inference to get a result.