Still illegal federally. A DEA agent still has the legal authority to arrest you (although, I doubt a jury would actually convict you). A non-citizen caught smoking weed can be denied citizenship if they try to naturalize. And they could face deportation proceedings. Even if you naturalize, if they found out you lied about the smoking weed, the administration could try to denaturalize you, especially this current administration would definitely attempt to.