Discovered this paper via a bit too vague MIT news article [1] Code examples are in their GitHub repo [2]

[1] https://news.mit.edu/2025/making-ai-generated-code-more-accu...

[2] https://github.com/genlm/genlm-control