Neural network training with acceleration

A method, comprising:

• storing, by a first computational storage device of one or more computational storage devices of a computational storage cluster, an embedding table;
• receiving, by the first computational storage device, an index vector comprising a first index and a second index;
• and
• calculating, by the first computational storage device, an embedded vector based on:
• a first row of the embedding table, corresponding to the first index, and
• a second row of the embedding table, corresponding to the second index,
• wherein:
• the computational storage cluster is connected to a graphics processing unit cluster by a cache-coherent system interconnect, and
• the graphics processing unit cluster comprises one or more graphics processing units.

A system, comprising:

• a graphics processing unit cluster;
• and
• a computational storage cluster connected to the graphics processing unit cluster by a cache-coherent system interconnect,
• wherein:
• the graphics processing unit cluster comprises one or more graphics processing units,
• the computational storage cluster comprises one or more computational storage devices,
• a first computational storage device of the one or more computational storage devices comprises persistent storage and means for processing, and
• the means for processing is configured to:
• store an embedding table;
• receive an index vector comprising a first index and a second index;
• and
• calculate an embedded vector based on:
• a first row of the embedding table, corresponding to the first index, and
• a second row of the embedding table, corresponding to the second index.