Hi Vivek,

Not exactly sure what you are looking for...

If you are trying to address input pipeline/CPU bottlenecks you might want to check out https://towardsdatascience.com/overcoming-data-preprocessing-bottlenecks-with-tensorflow-data-service-nvidia-dali-and-other-d6321917f851 and https://towardsdatascience.com/overcoming-ml-data-preprocessing-bottlenecks-with-grpc-ca30fdc01bee