If some of your players are on their phones a lot while you GM, maybe don’t punish them.

Maybe they are a little or a lot neurodivergent and don’t just focus on one thing easily and consistently like you do. Maybe that’s just how they engage socially.

Probably better to discuss things.