scib_metrics.nearest_neighbors.jax_approx_min_k#
- scib_metrics.nearest_neighbors.jax_approx_min_k(X, n_neighbors, recall_target=0.95, chunk_size=2048)[source]#
Run approximate nearest neighbor search using jax.
On TPU backends, this is approximate nearest neighbor search. On other backends, this is exact nearest neighbor search.
- Parameters:
- Return type: