docker build issue

Hello,

`bash docker_build_dependency_image.sh DEVICE=gpu JAX_VERSION=0.5.0` fails with the error below:

```15475.9 Collecting keras>=3.5.0 (from tensorflow>=2.13.0->-r requirements.txt (line 30))
15475.9 Downloading keras-3.7.0-py3-none-any.whl.metadata (5.8 kB)
25764.0 ERROR: Exception: 25764.0 Traceback (most recent call last):
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_internal/cli/base_command.py", line 106, in _run_wrapper 25764.0 status = _inner_run()
25764.0 ^^^^^^^^^^^^
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_internal/cli/base_command.py", line 97, in _inner_run
25764.0 return self.run(options, args)
25764.0 ^^^^^^^^^^^^^^^^^^^^^^^
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_internal/cli/req_command.py", line 67, in wrapper
25764.0 return func(self, options, args)
25764.0 ^^^^^^^^^^^^^^^^^^^^^^^^^
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_internal/commands/install.py", line 386, in run
25764.0 requirement_set = resolver.resolve(
25764.0 ^^^^^^^^^^^^^^^^^
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_internal/resolution/resolvelib/resolver.py", line 95, in resolve
25764.0 result = self._result = resolver.resolve(
25764.0 ^^^^^^^^^^^^^^^^^
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_vendor/resolvelib/resolvers.py", line 546, in resolve
25764.0 state = resolution.resolve(requirements, max_rounds=max_rounds)
25764.0 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
25764.0 File "/usr/local/lib/python3.12/dist-packages/pip/_vendor/resolvelib/resolvers.py", line 457, in resolve
25764.0 raise ResolutionTooDeep(max_rounds)
25764.0 pip._vendor.resolvelib.resolvers.ResolutionTooDeep: 200000
------
maxtext_gpu_dependencies.Dockerfile:47