Hey @Santosh-Gupta, if you’re looking for lazy loading for parquet files I’d recommend exploring https://github.com/vahidk/tfrecord 49 or https://github.com/uber/petastorm 74.

TFRecord would require you to go from parquet to tfrecord.
Petastorm should work out of the box.