Enhancing GPU Visibility for AI Workloads created with Kubeflow SDK
- 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
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