### Use Case / Motivation The current prediction model is based on PyTorch 1.0 and needs to be upgraded to the latest PyTorch version and used for inference with TensorRT. ### Proposed Implementation model training [repo](https://github.com/wheelos/prediction-training) ### Alternatives Considered _No response_
