Add typing friendly transformation interfaces#90

wdhongtw merged 1 commit into

maingoogle/torchax:mainfrom

transform-typinggoogle/torchax:transform-typingCopy head branch name to clipboard