Sentence-transformer

from sentence_transformers import SentenceTransformer

model = SentenceTransformer("all-MiniLM-L6-v2")

sentences = [

    "The weather is lovely today.",

    "It's so sunny outside!",

    "He drove to the stadium.",

]

embeddings = model.encode(sentences) # 3 x 384

model.similarity(embeddings, embeddings)