How to add comments, docstrings and keywrods to the torchscript extension?

Hello everyone,
Hope you are having a great day.
I’m building a Python extension using both Pybind11 and TorchScript extension as explained here. my job with the Pybinnd11 is done and now I’m trying the torch script extension and see how it fares against Pybind11.
I built one just now and it seems working however, I cant find anything regarding how to add docstring to the classes, methods, etc exposed using TorchScript. I also cant find the equivalent of Pybind11’s py::arg() which is used to add keywords , so on Python part, the keywords are visible and people can know what they are supposed to pass!

Any help in this regard is greatly appreciated.