Skip to main content

TPU index is a package for fast similarity search over large collections of high dimension vectors on Google Cloud TPUs

Project description

HitCount

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


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.6.tar.gz (2.9 kB view details)

Uploaded Source

Built Distribution

tpu_index-0.0.6-py3-none-any.whl (3.2 kB view details)

Uploaded Python 3

File details

Details for the file tpu_index-0.0.6.tar.gz.

File metadata

  • Download URL: tpu_index-0.0.6.tar.gz
  • Upload date:
  • Size: 2.9 kB
  • Tags: Source
  • Uploaded using 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

File hashes

Hashes for tpu_index-0.0.6.tar.gz
Algorithm Hash digest
SHA256 0e9430dbad89995ea44efe87129cb1b85343e402286d7927da837ab1820f7d77
MD5 2a319ac8fc948e3dae74ce5523b0b341
BLAKE2b-256 be0d02fb26a33f57ee1c5971da562185259aa8180cf047020cab5852c2bb3ed7

See more details on using hashes here.

File details

Details for the file tpu_index-0.0.6-py3-none-any.whl.

File metadata

  • Download URL: tpu_index-0.0.6-py3-none-any.whl
  • Upload date:
  • Size: 3.2 kB
  • Tags: Python 3
  • Uploaded using 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

File hashes

Hashes for tpu_index-0.0.6-py3-none-any.whl
Algorithm Hash digest
SHA256 ed6544c8a7995946959acbe6c398ecf782c48b241d41db42708699841aa02a66
MD5 dff41af8a5afa76c356df9aec8661f03
BLAKE2b-256 da82c2ea118622e638b215362a74dbcf246c9c83d4a1784e2f3a32c34e02cf71

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page