Recently, I switch to a role in ML and its been a change in my workflow so far - which is something I expected, but fun nonetheless. I don't get to do UI/UX in the visual sense but now I have to care about design of ML models and deployment so that other front-end clients can use them.
I would say give iOS development a go ahead because you want to!