Add map helpers for use in future melting pot version.
Add map helpers for use in future melting pot version.

PiperOrigin-RevId: 485877335
Change-Id: I852247fa319f3aefb0cf82a914df674254001e3b

deepmind/meltingpot,deepmind/meltingpot