Metadata-Version: 2.1
Name: tpu-index
Version: 0.0.6
Summary: TPU index is a package for fast similarity search over large collections of high dimension vectors on Google Cloud TPUs
Home-page: https://github.com/srihari-humbarwadi/tpu_index
Author: Srihari Humbarwadi
Author-email: sriharihumbarwadi97@gmail.com
License: UNKNOWN
Platform: UNKNOWN
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.5
Description-Content-Type: text/markdown
Requires-Dist: tensorflow (>=2.0.0)

[![HitCount](http://hits.dwyl.io/srihari-humbarwadi/tpu_index.svg)](http://hits.dwyl.io/srihari-humbarwadi/tpu_index)

## 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


