pytorch / functorch

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

Related projects

Alternatives and complementary repositories for functorch