[Feature]: Support tpu-checkpoint in snapshot agent

### Problem Statement

TPU support needs tpu-checkpoint integration in snapshot agent

### Proposed Solution

Add tpu-checkpoint support in snapshot agent.

### Alternatives Considered

_No response_

### Willingness to Contribute

Yes, I can submit a PR

### Additional Context

_No response_