I'll often use tensorflow probability's time series package.