DATA PARTITIONING IN AI TRAINING