Source code for the paper "Policy Architectures for Compositional Generalization in Control" - View it on GitHub