diff --git a/examples/mnist_classifier/requirements.in b/examples/mnist_classifier/requirements.in
index a8257fd70..a28dc21e4 100644
--- a/examples/mnist_classifier/requirements.in
+++ b/examples/mnist_classifier/requirements.in
@@ -5,5 +5,5 @@ flytekitplugins-deck-standard
 torch
 torchvision
 wandb
-numpy<1.22.0
-pandas<=1.4.3
+numpy
+pandas
diff --git a/examples/mnist_classifier/requirements.txt b/examples/mnist_classifier/requirements.txt
index 93fc31a85..0beabd880 100644
--- a/examples/mnist_classifier/requirements.txt
+++ b/examples/mnist_classifier/requirements.txt
@@ -18,8 +18,12 @@ aioitertools==0.11.0
     # via aiobotocore
 aiosignal==1.3.1
     # via aiohttp
-arrow==1.2.2
-    # via jinja2-time
+annotated-types==0.6.0
+    # via pydantic
+appdirs==1.4.4
+    # via wandb
+arrow==1.3.0
+    # via cookiecutter
 asttokens==2.4.1
     # via stack-data
 async-timeout==4.0.3
@@ -28,7 +32,7 @@ attrs==23.2.0
     # via
     #   aiohttp
     #   visions
-azure-core==1.26.3
+azure-core==1.30.0
     # via
     #   adlfs
     #   azure-identity
@@ -37,7 +41,7 @@ azure-datalake-store==0.0.53
     # via adlfs
 azure-identity==1.15.0
     # via adlfs
-azure-storage-blob==12.16.0
+azure-storage-blob==12.19.0
     # via adlfs
 binaryornot==0.4.4
     # via cookiecutter
@@ -45,7 +49,7 @@ botocore==1.34.34
     # via aiobotocore
 cachetools==5.3.2
     # via google-auth
-certifi==2021.10.8
+certifi==2024.2.2
     # via
     #   requests
     #   sentry-sdk
@@ -53,23 +57,25 @@ cffi==1.16.0
     # via
     #   azure-datalake-store
     #   cryptography
-chardet==4.0.0
+chardet==5.2.0
     # via binaryornot
-charset-normalizer==2.0.12
+charset-normalizer==3.3.2
     # via requests
-click==8.1.2
+click==8.1.7
     # via
     #   cookiecutter
     #   flytekit
     #   rich-click
     #   wandb
-cloudpickle==2.0.0
+cloudpickle==3.0.0
     # via flytekit
 comm==0.2.1
     # via ipywidgets
-cookiecutter==1.7.3
+contourpy==1.2.0
+    # via matplotlib
+cookiecutter==2.6.0
     # via flytekit
-croniter==1.3.4
+croniter==2.0.1
     # via flytekit
 cryptography==42.0.4
     # via
@@ -77,28 +83,30 @@ cryptography==42.0.4
     #   azure-storage-blob
     #   msal
     #   pyjwt
-cycler==0.11.0
+cycler==0.12.1
     # via matplotlib
 dacite==1.8.1
     # via ydata-profiling
-dataclasses-json==0.5.7
+dataclasses-json==0.5.9
     # via flytekit
 decorator==5.1.1
     # via
     #   gcsfs
     #   ipython
-diskcache==5.4.0
+diskcache==5.6.3
     # via flytekit
 docker==6.1.3
     # via flytekit
 docker-pycreds==0.4.0
     # via wandb
-docstring-parser==0.13
+docstring-parser==0.15
     # via flytekit
 exceptiongroup==1.2.0
     # via ipython
 executing==2.0.1
     # via stack-data
+filelock==3.13.1
+    # via torch
 flyteidl==1.10.7
     # via flytekit
 flytekit==1.10.7
@@ -107,7 +115,7 @@ flytekit==1.10.7
     #   flytekitplugins-deck-standard
 flytekitplugins-deck-standard==1.10.7
     # via -r requirements.in
-fonttools==4.33.2
+fonttools==4.49.0
     # via matplotlib
 frozenlist==1.4.1
     # via
@@ -119,11 +127,12 @@ fsspec==2024.2.0
     #   flytekit
     #   gcsfs
     #   s3fs
+    #   torch
 gcsfs==2024.2.0
     # via flytekit
-gitdb==4.0.9
+gitdb==4.0.11
     # via gitpython
-gitpython==3.1.27
+gitpython==3.1.42
     # via wandb
 google-api-core==2.17.1
     # via
@@ -155,15 +164,15 @@ googleapis-common-protos==1.62.0
     #   google-api-core
     #   grpcio-status
     #   protoc-gen-openapiv2
-grpcio==1.44.0
+grpcio==1.62.0
     # via
     #   flytekit
     #   grpcio-status
-grpcio-status==1.44.0
+grpcio-status==1.62.0
     # via flytekit
 htmlmin==0.1.12
     # via ydata-profiling
-idna==3.3
+idna==3.6
     # via
     #   requests
     #   yarl
@@ -171,7 +180,7 @@ imagehash==4.3.1
     # via
     #   visions
     #   ydata-profiling
-importlib-metadata==4.11.3
+importlib-metadata==7.0.1
     # via
     #   flytekit
     #   keyring
@@ -183,15 +192,15 @@ isodate==0.6.1
     # via
     #   azure-storage-blob
     #   flytekit
+jaraco-classes==3.3.1
+    # via keyring
 jedi==0.19.1
     # via ipython
-jinja2==3.1.1
+jinja2==3.1.3
     # via
     #   cookiecutter
-    #   jinja2-time
+    #   torch
     #   ydata-profiling
-jinja2-time==0.2.0
-    # via cookiecutter
 jmespath==1.0.1
     # via botocore
 joblib==1.3.2
@@ -202,17 +211,19 @@ jsonpickle==3.0.3
     # via flytekit
 jupyterlab-widgets==3.0.10
     # via ipywidgets
-keyring==23.5.0
+keyring==24.3.0
     # via flytekit
-kiwisolver==1.4.2
+kiwisolver==1.4.5
     # via matplotlib
+llvmlite==0.41.1
+    # via numba
 markdown==3.5.2
     # via flytekitplugins-deck-standard
 markdown-it-py==3.0.0
     # via rich
-markupsafe==2.1.1
+markupsafe==2.1.5
     # via jinja2
-marshmallow==3.15.0
+marshmallow==3.20.2
     # via
     #   dataclasses-json
     #   marshmallow-enum
@@ -225,7 +236,7 @@ marshmallow-jsonschema==0.13.0
     # via flytekit
 mashumaro==3.12
     # via flytekit
-matplotlib==3.5.1
+matplotlib==3.8.3
     # via
     #   -r requirements.in
     #   phik
@@ -236,6 +247,10 @@ matplotlib-inline==0.1.6
     # via ipython
 mdurl==0.1.2
     # via markdown-it-py
+more-itertools==10.2.0
+    # via jaraco-classes
+mpmath==1.3.0
+    # via sympy
 msal==1.26.0
     # via
     #   azure-datalake-store
@@ -251,15 +266,21 @@ multimethod==1.11.1
     # via
     #   visions
     #   ydata-profiling
-mypy-extensions==0.4.3
+mypy-extensions==1.0.0
     # via typing-inspect
 networkx==3.2.1
-    # via visions
-numpy==1.21.6
+    # via
+    #   torch
+    #   visions
+numba==0.58.1
+    # via ydata-profiling
+numpy==1.25.2
     # via
     #   -r requirements.in
+    #   contourpy
     #   imagehash
     #   matplotlib
+    #   numba
     #   pandas
     #   patsy
     #   phik
@@ -274,7 +295,7 @@ numpy==1.21.6
     #   ydata-profiling
 oauthlib==3.2.2
     # via requests-oauthlib
-packaging==21.3
+packaging==23.2
     # via
     #   docker
     #   marshmallow
@@ -282,7 +303,7 @@ packaging==21.3
     #   msal-extensions
     #   plotly
     #   statsmodels
-pandas==1.3.5
+pandas==2.2.0
     # via
     #   -r requirements.in
     #   flytekitplugins-deck-standard
@@ -293,15 +314,13 @@ pandas==1.3.5
     #   ydata-profiling
 parso==0.8.3
     # via jedi
-pathtools==0.1.2
-    # via wandb
 patsy==0.5.6
     # via statsmodels
 pexpect==4.9.0
     # via ipython
 phik==0.12.4
     # via ydata-profiling
-pillow==9.1.0
+pillow==10.2.0
     # via
     #   imagehash
     #   matplotlib
@@ -312,10 +331,6 @@ plotly==5.19.0
     # via flytekitplugins-deck-standard
 portalocker==2.8.2
     # via msal-extensions
-poyo==0.5.0
-    # via cookiecutter
-promise==2.3
-    # via wandb
 prompt-toolkit==3.0.43
     # via ipython
 protobuf==4.25.3
@@ -329,13 +344,13 @@ protobuf==4.25.3
     #   wandb
 protoc-gen-openapiv2==0.0.1
     # via flyteidl
-psutil==5.9.0
+psutil==5.9.8
     # via wandb
 ptyprocess==0.7.0
     # via pexpect
 pure-eval==0.2.2
     # via stack-data
-pyarrow==6.0.1
+pyarrow==15.0.0
     # via flytekit
 pyasn1==0.5.1
     # via
@@ -345,8 +360,10 @@ pyasn1-modules==0.3.0
     # via google-auth
 pycparser==2.21
     # via cffi
-pydantic==1.10.14
+pydantic==2.6.1
     # via ydata-profiling
+pydantic-core==2.16.2
+    # via pydantic
 pygments==2.17.2
     # via
     #   flytekitplugins-deck-standard
@@ -356,10 +373,8 @@ pyjwt[crypto]==2.8.0
     # via
     #   msal
     #   pyjwt
-pyparsing==3.0.8
-    # via
-    #   matplotlib
-    #   packaging
+pyparsing==3.1.1
+    # via matplotlib
 python-dateutil==2.8.2
     # via
     #   arrow
@@ -367,23 +382,25 @@ python-dateutil==2.8.2
     #   croniter
     #   matplotlib
     #   pandas
-    #   wandb
-python-json-logger==2.0.2
+python-json-logger==2.0.7
     # via flytekit
-python-slugify==6.1.1
+python-slugify==8.0.4
     # via cookiecutter
 pytimeparse==1.1.8
     # via flytekit
-pytz==2022.1
-    # via pandas
-pywavelets==1.4.1
+pytz==2024.1
+    # via
+    #   croniter
+    #   pandas
+pywavelets==1.5.0
     # via imagehash
 pyyaml==6.0.1
     # via
+    #   cookiecutter
     #   flytekit
     #   wandb
     #   ydata-profiling
-requests==2.27.1
+requests==2.31.0
     # via
     #   azure-core
     #   azure-datalake-store
@@ -402,6 +419,7 @@ requests-oauthlib==1.3.1
     # via google-auth-oauthlib
 rich==13.7.0
     # via
+    #   cookiecutter
     #   flytekit
     #   rich-click
 rich-click==1.7.3
@@ -418,25 +436,19 @@ scipy==1.11.4
     #   ydata-profiling
 seaborn==0.12.2
     # via ydata-profiling
-sentry-sdk==1.5.10
+sentry-sdk==1.40.5
     # via wandb
-setproctitle==1.2.3
-    # via wandb
-shortuuid==1.0.8
+setproctitle==1.3.3
     # via wandb
 six==1.16.0
     # via
     #   asttokens
     #   azure-core
-    #   cookiecutter
     #   docker-pycreds
-    #   grpcio
     #   isodate
     #   patsy
-    #   promise
     #   python-dateutil
-    #   wandb
-smmap==5.0.0
+smmap==5.0.1
     # via gitdb
 stack-data==0.6.3
     # via ipython
@@ -444,17 +456,19 @@ statsd==3.3.0
     # via flytekit
 statsmodels==0.14.1
     # via ydata-profiling
+sympy==1.12
+    # via torch
 tangled-up-in-unicode==0.2.0
     # via visions
 tenacity==8.2.3
     # via plotly
 text-unidecode==1.3
     # via python-slugify
-torch==1.11.0
+torch==2.2.0
     # via
     #   -r requirements.in
     #   torchvision
-torchvision==0.12.0
+torchvision==0.17.0
     # via -r requirements.in
 tqdm==4.66.2
     # via ydata-profiling
@@ -464,22 +478,27 @@ traitlets==5.14.1
     #   ipython
     #   ipywidgets
     #   matplotlib-inline
-typeguard==2.13.3
+typeguard==4.1.5
     # via ydata-profiling
-typing-extensions==4.2.0
+types-python-dateutil==2.8.19.20240106
+    # via arrow
+typing-extensions==4.9.0
     # via
     #   azure-core
     #   azure-storage-blob
     #   flytekit
     #   mashumaro
     #   pydantic
+    #   pydantic-core
     #   rich-click
     #   torch
-    #   torchvision
+    #   typeguard
     #   typing-inspect
-typing-inspect==0.7.1
+typing-inspect==0.9.0
     # via dataclasses-json
-urllib3==1.26.9
+tzdata==2024.1
+    # via pandas
+urllib3==1.26.18
     # via
     #   botocore
     #   docker
@@ -488,23 +507,26 @@ urllib3==1.26.9
     #   sentry-sdk
 visions[type-image-path]==0.7.5
     # via ydata-profiling
-wandb==0.12.15
+wandb==0.16.3
     # via -r requirements.in
 wcwidth==0.2.13
     # via prompt-toolkit
 websocket-client==1.7.0
     # via docker
-wheel==0.37.1
+wheel==0.42.0
     # via -r requirements.in
 widgetsnbextension==4.0.10
     # via ipywidgets
 wordcloud==1.9.3
     # via ydata-profiling
-wrapt==1.14.0
+wrapt==1.16.0
     # via aiobotocore
 yarl==1.9.4
     # via aiohttp
-ydata-profiling==4.5.1
+ydata-profiling==4.6.4
     # via flytekitplugins-deck-standard
-zipp==3.8.0
+zipp==3.17.0
     # via importlib-metadata
+
+# The following packages are considered to be unsafe in a requirements file:
+# setuptools