[WIP] Add support for torch.compile

amazon-science / chronos-forecasting

Chronos: Pretrained (Language) Models for Probabilistic Time Series Forecasting

https://arxiv.org/abs/2403.07815

Apache License 2.0

2.38k stars 273 forks source link

Closed abdulfatir closed 6 months ago

abdulfatir commented 6 months ago

Description of changes: This PR adds an option to torch.compile the generate function.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

lostella commented 6 months ago

Could the user not just torch.compile(pipeline.predict) instead?

abdulfatir commented 6 months ago

Yeah, actually I am not sure how helpful this is even.