A system including a memory device with microbumps are disclosed. A non-volatile memory device stores data for a machine learning operation. The non-volatile memory device comprises a set of microbumps. The set of microbumps are to transmit the data for the machine learning operation from the non-volatile memory device to another set of microbumps of a machine learning processing device that performs the machine learning operation.