pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.
1,395Updated this week

Related projects

Alternatives and complementary repositories for functorch