diff --git a/Pipfile b/Pipfile
index 02ab5a5..7cff69c 100644
--- a/Pipfile
+++ b/Pipfile
@@ -4,7 +4,6 @@ url = "https://pypi.org/simple"
 verify_ssl = true
 
 [dev-packages]
-python-language-server = {extras = ["all"],version = "*"}
 ipython = "*"
 matplotlib = "*"
 ptvsd = "*"
@@ -16,8 +15,8 @@ tensorflow-datasets = "*"
 black = "*"
 
 [packages]
-tensorflow = "==2.2.0"
-tensorflow-probability = "==0.10.1"
+tensorflow = "==2.4.0"
+tensorflow-probability = "==0.12.2"
 pandas = "*"
 matplotlib = "*"
 tensorflow-datasets = "*"
diff --git a/dockerfile b/dockerfile
index 2ffa173..e83e012 100644
--- a/dockerfile
+++ b/dockerfile
@@ -1,4 +1,4 @@
-FROM tensorflow/tensorflow:2.1.0-gpu-py3
+FROM tensorflow/tensorflow:2.4.0-gpu
 RUN apt-add-repository ppa:fish-shell/release-3
 RUN apt-get update
 RUN apt-get install -y zsh tmux wget
diff --git a/requirements.txt b/requirements.txt
index 04322d7..05288ec 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,88 +1,67 @@
+#
+# These requirements were autogenerated by pipenv
+# To regenerate from the project's Pipfile, run:
+#
+#    pipenv lock --requirements
+#
+
 -i https://pypi.org/simple
-absl-py==0.9.0
-astor==0.8.1
-attrs==19.3.0
-backcall==0.1.0
-bleach==3.1.4
-cachetools==4.1.0
-certifi==2020.4.5.1
-chardet==3.0.4
-cloudpickle==1.3.0
+absl-py==0.12.0
+astunparse==1.6.3
+attrs==20.3.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+cachetools==4.2.2; python_version ~= '3.5'
+certifi==2020.12.5
+chardet==4.0.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+cloudpickle==1.6.0; python_version >= '3.5'
 cycler==0.10.0
-decorator==4.4.2
-defusedxml==0.6.0
-entrypoints==0.3
-gast==0.2.2
-google-auth-oauthlib==0.4.1
-google-auth==1.14.1
+decorator==5.0.7; python_version >= '3.5'
+dill==0.3.3; python_version >= '2.6' and python_version != '3.0'
+dm-tree==0.1.6
+flatbuffers==1.12
+future==0.18.2; python_version >= '2.6' and python_version not in '3.0, 3.1, 3.2'
+gast==0.3.3; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+google-auth-oauthlib==0.4.4; python_version >= '3.6'
+google-auth==1.30.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4, 3.5'
 google-pasta==0.2.0
-grpcio==1.28.1
+googleapis-common-protos==1.53.0; python_version >= '3.6'
+grpcio==1.32.0
 h5py==2.10.0
-idna==2.9
-importlib-metadata==1.6.0 ; python_version < '3.8'
-ipykernel==5.2.1
-ipython-genutils==0.2.0
-ipython==7.13.0
-jedi==0.17.0
-jinja2==2.11.2
-json5==0.9.4
-jsonschema==3.2.0
-jupyter-client==6.1.3
-jupyter-core==4.6.3
-jupyterlab-server==1.1.1
-jupyterlab==2.1.0
-keras-applications==1.0.8
-keras-preprocessing==1.1.0
-kiwisolver==1.2.0
-markdown==3.2.1
-markupsafe==1.1.1
-matplotlib==3.2.1
-mistune==0.8.4
-nbconvert==5.6.1
-nbformat==5.0.6
-notebook==6.0.3
-numpy==1.18.3
-oauthlib==3.1.0
-opt-einsum==3.2.1
-pandas==1.0.3
-pandocfilters==1.4.2
-parso==0.7.0
-pexpect==4.8.0 ; sys_platform != 'win32'
-pickleshare==0.7.5
-prometheus-client==0.7.1
-prompt-toolkit==3.0.5
-protobuf==3.11.3
-ptyprocess==0.6.0 ; os_name != 'nt'
+idna==2.10; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+importlib-metadata==4.0.1; python_version < '3.8'
+importlib-resources==5.1.2; python_version < '3.9'
+keras-preprocessing==1.1.2
+kiwisolver==1.3.1; python_version >= '3.6'
+markdown==3.3.4; python_version >= '3.6'
+matplotlib==3.4.1
+numpy==1.19.5; python_version >= '3.6'
+oauthlib==3.1.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+opt-einsum==3.3.0; python_version >= '3.5'
+pandas==1.2.4
+pillow==8.2.0; python_version >= '3.6'
+promise==2.3
+protobuf==3.15.8
 pyasn1-modules==0.2.8
 pyasn1==0.4.8
-pygments==2.6.1
-pyparsing==2.4.7
-pyrsistent==0.16.0
-python-dateutil==2.8.1
-pytz==2019.3
-pyzmq==19.0.0
+pyparsing==2.4.7; python_version >= '2.6' and python_version not in '3.0, 3.1, 3.2'
+python-dateutil==2.8.1; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2'
+pytz==2021.1
 requests-oauthlib==1.3.0
-requests==2.23.0
-rsa==4.0
-scipy==1.4.1 ; python_version >= '3'
-send2trash==1.5.0
-sentencepiece==0.1.85
-six==1.14.0
-tensorboard==2.1.1
-tensorflow-addons==0.9.1
-tensorflow-estimator==2.1.0
-tensorflow-probability==0.9.0
-tensorflow==2.1.0
+requests==2.25.1; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+rsa==4.7.2; python_version >= '3.6'
+six==1.15.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2'
+tensorboard-data-server==0.6.0; python_version >= '3.6'
+tensorboard-plugin-wit==1.8.0
+tensorboard==2.5.0; python_version >= '2.7' and python_version not in '3.0, 3.1'
+tensorflow-datasets==4.2.0
+tensorflow-estimator==2.4.0
+tensorflow-metadata==0.30.0; python_version >= '3.6' and python_version < '4'
+tensorflow-probability==0.12.2
+tensorflow==2.4.0
 termcolor==1.1.0
-terminado==0.8.3
-testpath==0.4.4
-tornado==6.0.4
-traitlets==4.3.3
-typeguard==2.7.1
-urllib3==1.25.9
-wcwidth==0.1.9
-webencodings==0.5.1
-werkzeug==1.0.1
-wheel==0.34.2 ; python_version >= '3'
+tqdm==4.60.0; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3'
+typing-extensions==3.7.4.3; python_version < '3.8' and python_version < '3.8'
+urllib3==1.26.4; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4' and python_version < '4'
+werkzeug==1.0.1; python_version >= '2.7' and python_version not in '3.0, 3.1, 3.2, 3.3, 3.4'
+wheel==0.36.2; python_version >= '3'
 wrapt==1.12.1
-zipp==3.1.0
+zipp==3.4.1; python_version < '3.8'