Add a doc for working with impure functions#64

jialei777 merged 2 commits into

google:maingoogle/torchax:mainfrom

qihqi:qihqi-docsqihqi/torchax:qihqi-docsCopy head branch name to clipboard