[POC][functorch] use public PyTree API in torch.func
#137884
Closed
XuehaiPan wants to merge 72 commits intogh/XuehaiPan/182/basefrom
Closed
[POC][functorch] use public PyTree API in `torch.func`#137884XuehaiPan wants to merge 72 commits intogh/XuehaiPan/182/basefrom
XuehaiPan wants to merge 72 commits intogh/XuehaiPan/182/basefrom