functorch is JAX-like composable function transforms for PyTorch.
Here you can share your experience with the project you are suggesting or its comparison with functorch. Optional.
A valid email to send you a verification link when necessary or log in.
functorch