pytorch

119f5ca1 - Update on "[export] Initial serialization"

Update on "[export] Initial serialization" Serialization TODOs: - [ ] pytree spec - [ ] higher order ops - [ ] node metadata (specifically nn_module_stack/source_fn) - [ ] shape env - [ ] graph module metadata? [ghstack-poisoned]