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
1 maintainer
Metadata
1 maintainer
Release files for tpu-index 0.0.6
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| tpu_index-0.0.6.tar.gz | 2.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| tpu_index-0.0.6-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 6.1 kB
Release files / tpu_index-0.0.6.tar.gz
| Download URL | tpu_index-0.0.6.tar.gz |
|---|---|
| Size | 2.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
0e9430dbad89995ea44efe87129cb1b85343e402286d7927da837ab1820f7d77
|
|
BLAKE2b-256 checksum How to use checksums |
be0d02fb26a33f57ee1c5971da562185259aa8180cf047020cab5852c2bb3ed7
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.0 requests-toolbelt/0.9.1 tqdm/4.48.0 CPython/3.8.5
|
Release files / tpu_index-0.0.6-py3-none-any.whl
| Download URL | tpu_index-0.0.6-py3-none-any.whl |
|---|---|
| Size | 3.2 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
ed6544c8a7995946959acbe6c398ecf782c48b241d41db42708699841aa02a66
|
|
BLAKE2b-256 checksum How to use checksums |
da82c2ea118622e638b215362a74dbcf246c9c83d4a1784e2f3a32c34e02cf71
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.2.0 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.1.0 requests-toolbelt/0.9.1 tqdm/4.48.0 CPython/3.8.5
|
