Oh yes, this is something that we are interested in as well.
The new beam search should make this possible, although you will have to modify the code.
In particular see this function:
It will allow you to return a
(batchSize * beamSize) tensor at each step of beam search. If you return false on “bad” outputs it will ignore them.