docker build issue

[sfvaroglu]

Hello,

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

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