Towards Generalized Control: On-the-Fly In-Topic Generation Code for paper Towards Generalized Control: On-the-Fly In-Topic Generation. This is joint work with Jiatong Yu and served as the final project for COS597G at Princeton.