tpu_index is a package for fast similarity search over large collections of high dimension vectors on TPUs
Project description
TPU Index
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
Installation
!pip install tpu-index
Basic usage
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)
ToDo:
- Add more distance metrics
- Optional GPU support
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
tpu_index-0.0.3.tar.gz
(2.5 kB
view hashes)
Built Distribution
Close
Hashes for tpu_index-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2585e96cda13a78c651cddb2f7f4084e4f524c058856a4a0fd022261006bfc30 |
|
MD5 | d250f7c252f7d2dccd0b7712f60c82de |
|
BLAKE2b-256 | bee4afcce2819fc4f795ac780d14c36d8336779ed3d9d5240a7d960f66bdfef0 |