Abstract: A method of grouping embedding vectors in a storage device includes receiving, at a storage device, an inference query request including a first access index associated with a first embedding vector, determining, with a classification model, a first classification of the first access index, based on the first classification, searching for the first embedding vector in a cache of the storage device, receiving, at the storage device, a second access index associated with a second embedding vector, determining, with the classification model, a second classification of the second access index, and based on the second classification, searching for the second embedding vector in a persistent storage of the storage device.