[HN Gopher] Functorch: Jax-like composable function transforms f...
___________________________________________________________________
 
Functorch: Jax-like composable function transforms for PyTorch
 
Author : lnyan
Score  : 30 points
Date   : 2022-05-18 16:18 UTC (1 days ago)
 
web link (github.com)
w3m dump (github.com)
 
| 6gvONxR4sf7o wrote:
| I love jax, but I wish this kind of thing in python was easier to
| debug. The stacktraces you get using these patterns in this
| particular language can be wild.
| 
| For example, an unjitted function's error can be easier to debug
| in jax than in anything else, while the same error in the jitted
| function can be harder to debug in jax than in anything else. But
| it's similar with vmap, pmap, grad, and the rest of the
| transformations. Debugging gets nuts quickly.
| 
| But I don't think there's any way around that with these kinds of
| transformations in python, is there?
 
___________________________________________________________________
(page generated 2022-05-19 23:01 UTC)