What is the status of `functorch`?

1

The official docs say that functorch is deprecated and torch.func should be used instead.

However, functorch has recently received new features like functorch.dim and functorch.einops (as of PyTorch 2.1) that are not a part of torch.func. So I am a bit confused: what is the current relation between functorch and torch.func? If I find features in functorch useful, can I use them or I should wait for them being merged to torch.func ?

2

This seems like a relevant question! cc @ptrblck :pray:

3

I don’t see any recent commits to the main branch of pytorch/functorch as seen here so would stick to the docs and use torch.func in recent PyTorch versions.

2 Likes