Retry mechanism

I'd love to have a mechanism that retries failed tests but not with the rerun file mechanism.
I'm interested to have a report that is OK if at least one of the X attempts is OK (retry a test only when it fails indeed), and generate a unique report files. (Better again if the report can flag the flaky tests.)

Is something already available for this use case?

If no, I think about forking this repository to be able to implement my own AndroidFeatureRunner + RunNotifier, to be able to call child.run(myNotifier) as much as I need. Is it a good way to approach the problem?

Thanks for your time and this great tool!