US 11,734,205 B2
Parallel iterator for machine learning frameworks
Jacob Mulamootil Jacob, Cedar Park, TX (US); Gaurav Sanjay Ramdasi, Pune (IN); and Nabeel Meeramohideen Mohamed, Round Rock, TX (US)
Assigned to Micron Technology, Inc., Boise, ID (US)
Filed by MICRON TECHNOLOGY, INC., Boise, ID (US)
Filed on Jan. 11, 2022, as Appl. No. 17/573,545.
Application 17/573,545 is a continuation of application No. 16/722,408, filed on Dec. 20, 2019, granted, now 11,221,973.
Prior Publication US 2022/0138127 A1, May 5, 2022
This patent is subject to a terminal disclaimer.
Int. Cl. G06F 13/16 (2006.01); G06F 9/46 (2006.01); G06N 3/04 (2023.01); G06N 3/08 (2023.01)
CPC G06F 13/1668 (2013.01) [G06F 9/466 (2013.01); G06N 3/04 (2013.01); G06N 3/08 (2013.01)] 20 Claims
OG exemplary drawing
 
1. A method comprising:
receiving, from a machine learning (ML) framework executing on a host system, a request to retrieve data from a memory device of a memory sub-system, wherein the data comprises a plurality of logical partitions; and
initiating, based at least in part on a media type associated with the memory device, a set of parallel I/O threads to retrieve the data from the memory device, wherein each parallel I/O thread of the set of parallel I/O threads retrieves a different portion of the data from a different corresponding logical partition of the plurality of logical partitions and stores the different portion of the data in a I/O buffer of a set of I/O buffers corresponding to the set of parallel I/O threads in parallel.