GitHub repository: BlueMirrors/torchtrtz
Primary language: Python
Description: Pytorch models built and saved to generate tensorrt weights file. Sibling project to  use the generated trt weights file and create tensorrt engine using TensorRT C++ api.