Hi,
I suggest to take a look at this repo: GitHub - mandarjoshi90/coref: BERT for Coreference Resolution
It includes multiple models (BERT, SpanBERT) fine-tuned on OntoNotes, an important benchmark for coreference resolution.
There’s also a demo notebook, showcasing how to run inference for a new piece of text to find all entity clusters.