I believe it's only supported on 3.10 IIRC, you can try updating pip and see if it works then, i did some digging and most of what I'm finding seemed to be the solution
https://github.com/tensorflow/tfx/issues/5930