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 ?