I'd love to read your PhD thesis and papers! I'm also an AI researcher, currently doing a Master's in something else, but compositionality and representation learning is very interesting to me.