Fix: Use without_apply_rng in MNIST GAN example.
Fix: Use without_apply_rng in MNIST GAN example.

PiperOrigin-RevId: 321563278
Change-Id: Id21f5b09081f240b257a925297036a725fab8e3f

deepmind/dm-haiku,deepmind/dm-haiku