IDE Setup: linking forward and __call__

Do any pytorch developers/users have a python IDE configuration (or other tooling) that allows them to easily jump between invocations of a model and the definition of model.forward? I am in pycharm, where “show usages”, “jump to caller” etc. work for most class methods but not this, but happy to switch to VSCODE or another IDE.

Thanks!