It looks like some of the formatting of the docstrings in tf v2 don't quite match the formats used in keras which results in ugly rendering: 