kubeflow / kubeflow/sdk

Enhancing GPU Visibility for AI Workloads created with Kubeflow SDK

Open
#165 17 comments 4 reactions 0 assignees View on GitHub
area/observability kind/discussion kind/feature
Dominant language
Python
Stars
148
Forks
262
Avg merge
1d 2h
Merged PRs (30d)
1

Description

### What you would like to be added?

During KubeCon + CloudNativeCon NA 2025, several users reached out to discuss challenges around GPU failures and the need for better observability of GPU utilization when running AI workloads.

A few potential approaches were proposed, including:
1. Deeper integration with the PyTorch Profiler to capture per–neural-network-layer GPU utilization: https://docs.pytorch.org/tutorials/recipes/recipes/profiler_recipe.html
2. Providing optional SSH access to the MASTER node so users can run tools like nvidia-smi for real-time GPU monitoring (e.g. new `TrainerClient()` API)
3. Any other ideas ?

Let’s use this issue to brainstorm solutions that can improve visibility into GPU utilization and failure modes, enabling users to better tune their workloads and cluster configurations.

cc @kubeflow/kubeflow-sdk-team

### Why is this needed?

Improve GPU utilization of AI workloads.

### Love this feature?

Give it a 👍 We prioritize the features with most 👍

Contributor guide

Open the contributing guide

Research direction

No source files or tests are named. Start by reviewing the Kubeflow SDK's TrainerClient entry point and compare the proposed PyTorch Profiler integration with optional SSH access to the MASTER node. Done would require a decided approach and a defined way to observe GPU utilization and failure modes, which this issue does not yet specify.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
ai, machine-learning, observability
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Quiet
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.