Issue importing functorch

I tried importing functorch and torch in python 3.9.7

import torch
from functorch import vmap, grad

and ran into a circular import issue.

ImportError                               Traceback (most recent call last)
/var/folders/4p/5v2n8wpn58v6xw674tjlss680000gn/T/ipykernel_64234/ in <module>
      1 import torch
----> 2 from functorch import vmap, grad

/usr/local/lib/python3.11/site-packages/functorch/ in <module>
      5 # LICENSE file in the root directory of this source tree.
      6 import torch
----> 7 from . import _C
      9 # Top-level APIs. Please think carefully before adding something to the

ImportError: cannot import name '_C' from partially initialized module 'functorch' (most likely due to a circular import) (/usr/local/lib/python3.11/site-packages/functorch/

Any way around it? Is this a bug in the packages?
Submitting this here since I could not find an easy fix

The functorch package has moved to core so you don’t need to have a seperate import instead you can use torch.func.vmap torch.func.vmap — PyTorch 2.0 documentation