Add test to ensure eval_summary and tabulate work under jit.
Add test to ensure eval_summary and tabulate work under jit.

PiperOrigin-RevId: 356240396
Change-Id: I87ed97c26f7592188b7317c93fa39224154de04e

deepmind/dm-haiku,deepmind/dm-haiku