Fully AI-based Data layer for training ML Models