There's a proposal up for AM upgrades https://issues.apache.org/jira/browse/YARN-4470
I've actually commented based on the work we had to do for AM restart and the work Gour did for slider rolling upgrades. Right now I'm not convinced the (fairly ambitious) design is needed, especially given how much of the RU update work went into dealing with container upgrade; AM update was a simple restart operation.