Improve NeMo Curator Experience for Pytorch Models (with crossfit)

[VibhuJawa]

Is your feature request related to a problem? Please describe.

Based on user feedback we need to fix the following to make user experience better:

• [x] Enable PyTorch to share same memory pool as RMM via cli #1392 PR in dask-cuda merged, will follow up for curator in next release
• [x] Fix GPU-id logging of crossfit
• [x] Enable memory estimation of crossfit+PyTorch when RMM backend is used

[Christina-Young-NVIDIA]

At risk because not all items above will complete in this sprint. Vibhu to break into two issues.