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

Overall goal would be to align caikit streams with torch / transformers data loader

Describe the solution you'd like

A clear and concise description of what you want to happen.

Describe alternatives you've considered

A clear and concise description of any alternative solutions or features you've considered.

Additional context

Add any other context about the feature request here.