Add a checkpointing config class to the distributed layout.
Add a checkpointing config class to the distributed layout.

PiperOrigin-RevId: 427435196
Change-Id: I133e2953618fd385ccbbb650be9442078b1d4874

deepmind/acme,deepmind/acme