I'm not sure if it's SOTA these days but we use ScaNN at my job. https://github.com/google-research/google-research/tree/mast...