facebookresearch/functorch

functorch is JAX-like composable function transforms for PyTorch.

4 best points · 0 comments · 1 Hacker News post · 1,433 GitHub stars · View on GitHub

Hacker News discussions

• Functorch: Jax-like composable function transforms for PyTorch (4 points, 0 comments, by reimbar, 2021-09-23)