A system includes a memory device to maintain data for a machine learning operation. The memory device includes solder balls. The system further includes a machine learning processing device to perform the machine learning operation. The system further includes a processing device to select, based on the machine learning operation, a set of solder balls from the plurality of solder balls to transmit the data from the non-volatile memory device to the machine learning processing device.