pytorch/functorch
View on GitHub ↗ ★ 1,433 Jupyter Notebook archived
functorch is JAX-like composable function transforms for PyTorch.
4 mentions on Hacker News by 4 people: 1 story, 3 comments · first 2021-10-22 · latest 2022-05-18
20082026
-
…[1] https://docs.hylang.org/en/alpha/ [2] https://github.com/google/jax [3] https://github.com/pytorch/functorch
comment by jszymborski · 2021-11-26 · in PyTorch: Where we are headed and why it looks a lot like Julia (but not exactly) -
…in practice. Fwiw, there's an offshoot of pytorch that is aiming to provide this sort of API (see https://github.com/pytorch/functorch and look at eager_compilation.py). (Disclaimer: I worked on this until quite recently.)
comment by bertr4nd · 2021-11-06 · in Show HN: How does JAX allocate memory on a TPU? An interactive C++ walkthrough -
…(related to its function transformations) I'd say that Pytorch is moving towards the first (see https://github.com/pytorch/functorch ) but not the second. Disclaimer: I work on PyTorch, and Functorch more specifically, although my opinions…
comment by chillee · 2021-10-22 · in PyTorch 1.10