Though we are initially looking at a Neural approach for this, ChatGPT might be able to help us scale this well to out of distribution scenarios by using a neural-symbolic approach. Would be thrilled to talk more about this. Check out