There are some changes, but the whole system works much as it does in the live game today. The same 4 Keywardens are still the source of keys, and the KWs are still randomly located on their same levels.
See the wiki DiabloWikiInfernal Machine article for all the details on where they are. You still have to scour their whole levels to find them, and though that’s not any harder than it once was,
it feels harder since 1) none of those levels are super EXP farming areas anymore that you’d be doing anyway, and 2) RoS has trained us like dolphins at Sea World to seek out the blinking yellow sonar circle on the map when racing for Bounties. KWs are never bounties, and thus it *feels* long and wandering to hunt for them.