"""Token sampler."""
# TODO