ERROR: Package 'jax' requires a different Python: 3.8.10 not in '>=3.9'

When installing the DFC on Ubuntu 20.04, Python 3.8 Getting the above error message.
A simple solution is to update pip:
pip install --upgrade pip
then continue with the DFC installation.