Merge branch 'main' into sbosisio/support_axlearn

Author: Steboss

.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 GIT_USER_EMAIL=jax@nvidia.com
 ARG CLANG_VERSION=18
@@ -42,6 +42,7 @@ apt_packages=(
   vim
   wget
   jq
+  zip
   # llvm.sh
   lsb-release 
   software-properties-common