How are the tax rules different between Roth IRA for Non-Resident & Resident Alian?

Is it better to have Roth IRA, 401k or both?
As I know, there is 30% taxes up front for Non-resident alien to have 401k.

What else should I know?

Thank you