3 ms·
The underlying concept of JAX, function transformations, is very powerful and elegant. PyTorch 2.0 has gotten a similar underlying feature for torch.compile no
by albertzeyer 4y ago
The underlying concept of JAX, function transformations, is very powerful and elegant.
PyTorch 2.0 has gotten a similar underlying feature for torch.compile now. https://pytorch.org/get-started/pytorch-2.0/ https://pytorch.org/get-started/pytorch-2.0/