Question on who can invest in 529 plan

I am a US Permanent Resident and have a boy. (He is US citizen). We are considering going back to India, and hence I may not keep the US Permanent resident status. Can I invest in 529 plan for my son and still control if not being a US resident.

Thanks.
K