Scaling ML embedding models to serve a billion queries