Add a simple test for beam search to perform inference in 2-state Markov model.
Add a simple test for beam search to perform inference in 2-state Markov model.

PiperOrigin-RevId: 427551125

google-research/t5x