/flax-sentence-embeddings

Shared code for training sentence embeddings with Flax / JAX

flax-sentence-embeddings

This repository will be used to share code for the Flax / JAX community event to train sentence embeddings on 1B+ training pairs.

You can add your code by creating a pull request.