[AIR/Data] Revamping APIs for Offline Batch Inference (#25)

Signed-off-by: amogkam <[email protected]>
Signed-off-by: Amog Kamsetty <[email protected]>
Co-authored-by: Eric Liang <[email protected]>

Author: amogkam