Match the world model config in serialized/non-serialized policy experiments.
Match the world model config in serialized/non-serialized policy experiments.

PiperOrigin-RevId: 275481127

google/trax,google/trax