Is it a good idea to invest in 401(k) while on H1b when the long-term stay in US isn't planned/decided yet? Your opinions and experiences are appreciated!

Thanks,

delicate