Code for Learning to Synthesize Programs as Interpretable and Generalizable Policies in NeurIPS 2021 - View it on GitHub