How To Add NMS in pytorch model so that it gets converted into TFLITE

First off, thanks for this amazing repo. Im working on a ssd model in pytorch and i want to add post processing (NMS) into the tflite, how can i add it into my model so that it gets translated to tflite's NMS OP. thanks

The issue concerns adding SSD post-processing and NMS conversion from a PyTorch model into TFLite, but it names no repository files, tests, or entry points. Start by locating the model-conversion path and any existing post-processing support; completion would require a documented, working conversion to the requested TFLite NMS operation.

Written by the indexing model from the issue text.