Add remote tensorflow profiling to training jobs. #519
Closed
Description
To support on demand profiling of tensorflow , add handlers for profiling requests when running training jobs.
- User updates their code to import the training util for profiling.
- User submits a training job to Vertex AI.
- Through Vertex AI Tensorboard, user can profile their remote training job through the "Capture Profile" functionality in Tensorboard.