kubeflow / kubeflow/trainer

Transparent GPU Checkpointing of TrainJobs with CRIU

Open
#2,777 15 comments 7 reactions 1 assignee Claimed by @superasymmetry View on GitHub
area/controller kind/feature
Dominant language
Go
Stars
2.2k
Forks
1.1k
Avg merge
3d 22h
Merged PRs (30d)
39

Description

### What you would like to be added?

As part of the recent Training WG call, we discussed transparent GPU checkpointing for AI workloads using CRIU.
Recordings: https://youtu.be/j6BR7VEcWPY?t=559

The initial discussions started in this issue, but we want to separate the exporter Job from the transparent checkpointing functionality: https://github.com/kubeflow/trainer/issues/2245
Let’s use this issue to track progress and explore how we can integrate this capability with Kubeflow Trainer.

Resources:
- KubeCon London talks: https://youtu.be/BSoEY_tpxIo and https://youtu.be/3oWODC2mdk0
- CRIU Coordinator: https://github.com/checkpoint-restore/criu-coordinator
- Checkpointing ctl: https://github.com/checkpoint-restore/checkpointctl

cc @kubeflow/kubeflow-trainer-team @kramaranya @astefanutti @rst0git @adrianreber @shravan-achar @bigsur0

### Why is this needed?

Enhance TrainJob fault-tolerance and resiliency with transparent GPU checkpointing.

### Love this feature?

Give it a 👍 We prioritize the features with most 👍

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.