dump graph compiler - Githubissues

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Closed yushangdi closed 1 year ago

yushangdi commented 1 year ago

add a compiler function to dump raw graphs

yushangdi commented 1 year ago