Move to CUDA DL 25.02 base container (#1320)

- Update patch for nsys 2025.1.1
- Install zip for a JAX unit test

Author: olupton

.github/container/Dockerfile.base

@@ -1,5 +1,5 @@
 # syntax=docker/dockerfile:1-labs
-ARG BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:25.01-cuda12.8-devel-ubuntu24.04
+ARG BASE_IMAGE=nvcr.io/nvidia/cuda-dl-base:25.02-cuda12.8-devel-ubuntu24.04
 ARG GIT_USER_NAME="JAX Toolbox"
 ARG [email protected]
 ARG CLANG_VERSION=18
@@ -42,6 +42,7 @@ apt_packages=(
   vim
   wget
   jq
+  zip
   # llvm.sh
   lsb-release 
   software-properties-common