Check if the method can be made more efficient for large datasets https://github.com/amazon-science/chronos-forecasting/blob/7daaa7194c9472f16eab0ea2f9d7eac317002201/src/chronos/df_utils.py#L212