approximate-nearest-neighbor-graphs/requirements.txt at main · Damianzoub/approximate-nearest-neighbor-graphs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# Python dependencies for ANNS Thesis
# Generated: 2024
# Python >= 3.8 required

# Core scientific computing
numpy==1.24.3
pandas==2.0.3
scipy==1.11.1

# Visualization
matplotlib==3.7.2
seaborn==0.12.2

# Machine Learning / Embeddings
torch==2.0.1
torchvision==0.15.2
scikit-learn==1.3.0
sentence-transformers==2.2.2

# ANN Libraries
faiss-cpu==1.7.4
hnswlib==0.7.0

# Data formats
h5py==3.9.0

# Configuration and reproducibility
pyyaml==6.0.1

# Benchmarking and profiling
psutil==5.9.5
memory-profiler==0.61.0


# Utilities
tqdm==4.65.0
tabulate==0.9.0

# Testing (optional)
pytest==7.4.0
pytest-cov==4.1.0
ollama==0.16.3
chromadb
# in case wanting to run the "real-world-scenarios"
# ones download ollama chromadb
# in terminal:
# ollama pull nomic-embed-text
# ollama pull llama3.2