NVIDIA Time Series Prediction Platform

NVIDIA Time Series Prediction Platform

NVIDIA Time Series Prediction Platform (TSPP) is a tool designed to compare easily and experiment with arbitrary combinations of forecasting models, time-series datasets, and other configurations. The NVIDIA-TSPP provides functionality to explore the hyperparameter search space, run accelerated model training using distributed training and Automatic Mixed Precision (AMP), and deploy and run inference on accelerated model formats on the NVIDIA Triton Inference Server. The notebooks cover the following topics:

1. NVDIA-TSPP Overview
2. Triton Inference
3. Add a new Dataset
4. Multi-GPU Training and Hyper Parameter Search

NVIDIA-TSPP official release is available on the Nvidia Deep Learning Examples github. Our blog presents the technical overview of the platform.

Get Started with Training

To help you get started, we have created sample Jupyter Notebooks that can be easily deployed on Vertex AI using NGC’s One Click Deploy feature. This feature automatically sets up the Vertex AI instance with an optimal configuration, preloads the dependencies, runs the software from NGC without any need to set up the infrastructure.

Simply click on the button that reads “Deploy to Vertex AI” and follow the instructions.

Note: A customized kernel for the Jupyter Notebook is used as the primary mechanism for deployment. This kernel has been built on the Pytorch Container. For more information on the container itself, please refer to this link for more information: NVIDIA-TSPP github