pytorch

946d4fe3 - Update on "Add *_only and all/any pytree utilities"

Update on "Add *_only and all/any pytree utilities" With a sample usage in proxy tensor to show how they can shorten your code dramatically. Signed-off-by: Edward Z. Yang <ezyangfb.com> [ghstack-poisoned]