RecSim broken due to latest upgrade

Hi

While installing recsim we ran into following issue as it was trying latest version of dopamine-rl 3.1.1:

```
Collecting jaxlib>=0.1.52 (from dopamine-rl>=2.0.5->recsim)
Could not find a version that satisfies the requirement jaxlib>=0.1.52 (from dopamine-rl>=2.0.5->recsim) (from versions: 0.1, 0.1.1, 0.1.2, 0.1.3, 0.1.4, 0.1.5, 0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.1.11, 0.1.12, 0.1.13, 0.1.14, 0.1.15, 0.1.16, 0.1.17, 0.1.18, 0.1.19, 0.1.20, 0.1.21, 0.1.22, 0.1.23)
No matching distribution found for jaxlib>=0.1.52 (from dopamine-rl>=2.0.5->recsim)
```
**Steps to reproduce:**
`pip3 --default-timeout=60000 install --upgrade --no-cache-dir recsim --user`

**stack trace:**

```
Requirement already up-to-date: recsim in /usr/local/lib/python3.6/dist-packages
Collecting numpy (from recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/numpy/1.19.1/numpy-1.19.1-cp36-cp36m-manylinux1_x86_64.whl (13.4MB)
Requirement already up-to-date: absl-py in /usr/local/lib/python3.6/dist-packages (from recsim)
Collecting tensorflow (from recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/tensorflow/1.14.0/tensorflow-1.14.0-cp36-cp36m-manylinux1_x86_64.whl (109.2MB)
Requirement already up-to-date: gym in /usr/local/lib/python3.6/dist-packages (from recsim)
Collecting dopamine-rl>=2.0.5 (from recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/dopamine-rl/3.1.1/dopamine_rl-3.1.1-py3-none-any.whl (112kB)
Requirement already up-to-date: gin-config in /usr/local/lib/python3.6/dist-packages (from recsim)
Collecting scipy (from recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/scipy/1.5.2/scipy-1.5.2-cp36-cp36m-manylinux1_x86_64.whl (25.9MB)
Collecting six (from absl-py->recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/six/1.15.0/six-1.15.0-py2.py3-none-any.whl
Collecting wheel>=0.26 (from tensorflow->recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/wheel/0.34.2/wheel-0.34.2-py2.py3-none-any.whl
Requirement already up-to-date: grpcio>=1.8.6 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: tensorboard<1.15.0,>=1.14.0 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: tensorflow-estimator<1.15.0rc0,>=1.14.0rc0 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: wrapt>=1.11.1 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: protobuf>=3.6.1 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: termcolor>=1.1.0 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: keras-applications>=1.0.6 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: google-pasta>=0.1.6 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: keras-preprocessing>=1.0.5 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: gast>=0.2.0 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: astor>=0.6.0 in /usr/local/lib/python3.6/dist-packages (from tensorflow->recsim)
Requirement already up-to-date: pyglet<=1.5.0,>=1.4.0 in /usr/local/lib/python3.6/dist-packages (from gym->recsim)
Requirement already up-to-date: cloudpickle<1.4.0,>=1.2.0 in /usr/local/lib/python3.6/dist-packages (from gym->recsim)
Requirement already up-to-date: Pillow>=7.2.0 in /usr/local/lib/python3.6/dist-packages (from dopamine-rl>=2.0.5->recsim)
Collecting flax>=0.2.0 (from dopamine-rl>=2.0.5->recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/flax/0.2.0/flax-0.2.0-py3-none-any.whl (84kB)
Collecting tf-slim>=1.0 (from dopamine-rl>=2.0.5->recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/tf-slim/1.1.0/tf_slim-1.1.0-py2.py3-none-any.whl (352kB)
Collecting jax>=0.1.73 (from dopamine-rl>=2.0.5->recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/jax/0.1.73/jax-0.1.73.tar.gz (419kB)
Collecting pygame>=1.9.2 (from dopamine-rl>=2.0.5->recsim)
Downloading https://example.com/nexus/repository/pypi-all/packages/pygame/1.9.6/pygame-1.9.6-cp36-cp36m-manylinux1_x86_64.whl (11.4MB)
Requirement already up-to-date: opencv-python>=3.4.8.29 in /usr/local/lib/python3.6/dist-packages (from dopamine-rl>=2.0.5->recsim)
Collecting jaxlib>=0.1.52 (from dopamine-rl>=2.0.5->recsim)
Could not find a version that satisfies the requirement jaxlib>=0.1.52 (from dopamine-rl>=2.0.5->recsim) (from versions: 0.1, 0.1.1, 0.1.2, 0.1.3, 0.1.4, 0.1.5, 0.1.6, 0.1.7, 0.1.8, 0.1.9, 0.1.11, 0.1.12, 0.1.13, 0.1.14, 0.1.15, 0.1.16, 0.1.17, 0.1.18, 0.1.19, 0.1.20, 0.1.21, 0.1.22, 0.1.23)
No matching distribution found for jaxlib>=0.1.52 (from dopamine-rl>=2.0.5->recsim)
```

Temporary workaround is to downgrading `dopamine-rl = 3.0.1`.