3rd gen engine swap, 2ZR? 2AR? Swap newbie

[m.flores.0928]

I'm thinking about a motor swap. I have a 2014 4-door hatch with an auto. The current 1NZ isn't doing it for me, and as a few others have already done, I'm thinking about a 2ZR swap. I want to keep the auto trans, just to keep it easier.

Maybe a 2AR from a camry? Any thoughts on this?

As far as legalities and technicalities go, what sort of smog and registration laws do I have to follow? How would I make a swapped yaris smog and road legal?

As far as the mechanical installation goes, I am a moderate mechanic, I used to work in the field, but not anymore. I feel like I am capable of the installation. Sourcing parts, tools, figuring out the wiring and computer stuff, that's the easy part. I live in Sonoma County California. I aim to keep this a sleeper. I would like some modest suspension mods, maybe lower it, bigger sway bars, and a rear disc conversion, that sorta thing. That's about it.

I have never swapped an engine before, any and all tips and advice is welcome and appreciated! Thanks!