I highly recommend PyCharm, and it’s free if you are currently a student (they offer a free license to students).
But I don’t think it has specific PyTorch integration (i.e autocomplete for modules / submodules).