3 ms·
Tangentially related but there is an effort to get some of the features of JAX into PyTorch: https://pytorch.org/functorch/ https://pytorch.org/functorch/
by xico 4y ago
Tangentially related but there is an effort to get some of the features of JAX into PyTorch: https://pytorch.org/functorch/ https://pytorch.org/functorch/