Go into industry not academia. Academia won’t pay well and they do not appreciated good quality software engineering. For most labs, ML is considered a tool, not a primary driver of innovation. Look towards places like Isomorphic labs and similar companies that are cash rich and has a strong engineering heritage.