Linux Agent for NVIDIA GPU-Optimized Image for PyTorch is not auto upgradable

Linux Agent for NVIDIA GPU-Optimized Image for PyTorch is not auto upgradable. Current version of Linux agent is 2.2.46 which is not supported by Azure ( Extensions and virtual machine agent minimum version support - Virtual Machines | Microsoft Docs). We faced some issue while connecting to Azure VM with NVIDIA PyTorch image using Bastion host. Azure team confirmed that there is some endpoint restriction at OS level which is restricting to communicate. I didn’t find any support queue so posting my issue here. Thanks!