pytorch

2ea2860b - Add *_only and all/any pytree utilities

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 <ezyang@fb.com> [ghstack-poisoned]