This may be obvious and not what you want to do, but Lemmy's own UI is optimized for threaded conversations within a post. Browsing Lemmy communities from an account on a Lemmy instance looks pretty sensible.
That doesn't help you manage comment toots from Mastodon, but if you don't find another solution... it will look better from Lemmy itself.