TPU index is a package for fast similarity search over large collections of high dimension vectors on Google Cloud TPUs
TPU Index is a package for fast similarity search over large collections of high dimension vectors on TPUs. This package was built to support our project that we developed for https://tfworld.devpost.com/.
Link to our project: https://devpost.com/software/naturallanguagerecommendations
!pip install tpu-index
from tpu_index import TPUIndex index = TPUIndex(num_tpu_cores=8) index.create_index(vectors) # vectors = numpy array, shape == [None, None] ... D, I = index.search(xq, distance_metric='cosine', top_k=5)
- Add more distance metrics
- Optional GPU support
Release history Release notifications | RSS feed
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Hashes for tpu_index-0.0.6-py3-none-any.whl