ci: add dev container for easier development (#680)

This enables users to develop inside a prebuilt container. This can save
a lot of efforts of setting up cuda toolkit etc.
ref: https://code.visualstudio.com/docs/devcontainers/containers

.devcontainer/Dockerfile

@@ -0,0 +1,58 @@
+FROM nvidia/cuda:12.4.0-devel-ubuntu22.04
+
+# Update package lists and install system dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    git \
+    clang-format \
+    libibverbs-dev \
+    librdmacm-dev \
+    rdma-core \
+    libnuma-dev \
+    vim \
+    openmpi-bin \
+    libopenmpi-dev \
+    zsh \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install oh-my-zsh
+RUN sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended
+
+# Install powerlevel10k theme
+RUN git clone --depth=1 https://github.com/romkatv/powerlevel10k.git ${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/powerlevel10k
+
+# Install zsh-autosuggestions
+RUN git clone https://github.com/zsh-users/zsh-autosuggestions ${ZSH_CUSTOM:-~/.oh-my-zsh/custom}/plugins/zsh-autosuggestions
+
+# Configure zsh
+RUN sed -i 's/ZSH_THEME="robbyrussell"/ZSH_THEME="powerlevel10k\/powerlevel10k"/' ~/.zshrc && \
+    sed -i 's/plugins=(git)/plugins=(git zsh-autosuggestions)/' ~/.zshrc
+
+# Create a non-root user
+ARG USERNAME=devuser
+ARG USER_UID=1003
+ARG USER_GID=$USER_UID
+
+# Create the user
+RUN groupadd --gid $USER_GID $USERNAME \
+    && useradd --uid $USER_UID --gid $USER_GID -m $USERNAME \
+    # [Optional] Add sudo support
+    && apt-get update \
+    && apt-get install -y sudo \
+    && echo $USERNAME ALL=\(root\) NOPASSWD:ALL > /etc/sudoers.d/$USERNAME \
+    && chmod 0440 /etc/sudoers.d/$USERNAME \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy zsh configuration to the new user's home
+RUN cp -r /root/.oh-my-zsh /home/$USERNAME/.oh-my-zsh && \
+    cp /root/.zshrc /home/$USERNAME/.zshrc && \
+    chown -R $USERNAME:$USERNAME /home/$USERNAME/.oh-my-zsh && \
+    chown $USERNAME:$USERNAME /home/$USERNAME/.zshrc
+
+# Switch to non-root user
+USER $USERNAME
+WORKDIR /home/$USERNAME
+
+# Set zsh as default shell
+ENV SHELL=/bin/zsh
+CMD [ "zsh" ]

.devcontainer/devcontainer.json

@@ -0,0 +1,17 @@
+{
+    "name": "CUDA Development Container",
+    "build": {
+        "dockerfile": "Dockerfile",
+        "context": "."
+    },
+    "runArgs": [
+        "--gpus=all"
+    ],
+    "customizations": {
+        "vscode": {
+            "extensions": [
+            ]
+        }
+    },
+    "remoteUser": "devuser"
+}