usearch 2.10.3 → 2.11.0

package/simsimd/package.json

@@ -0,0 +1,61 @@
+{
+  "name": "simsimd",
+  "version": "4.2.2",
+  "description": "Fastest SIMD-Accelerated Vector Similarity Functions for x86 and Arm",
+  "homepage": "https://github.com/ashvardanian/simsimd",
+  "author": "Ash Vardanian",
+  "license": "Apache 2.0",
+  "keywords": [
+    "vector",
+    "distance",
+    "similarity",
+    "search",
+    "ai",
+    "math",
+    "avx2",
+    "avx512",
+    "neon",
+    "sve",
+    "arm",
+    "x86",
+    "simd",
+    "f16"
+  ],
+  "scripts": {
+    "install": "node-gyp-build",
+    "prebuild-single": "prebuildify --napi --strip --target=10.4.0",
+    "prebuild-arm64": "prebuildify --arch arm64 --napi --strip --target=10.4.0",
+    "prebuild-darwin-x64+arm64": "prebuildify --arch arm64+x64 --napi --strip --target=10.4.0",
+    "build-js": "rm -fr javascript/dist/* && tsc -p javascript/tsconfig-esm.json && tsc -p javascript/tsconfig-cjs.json && cp javascript/dist-package-esm.json javascript/dist/esm/package.json && cp javascript/dist-package-cjs.json javascript/dist/cjs/package.json",
+    "test": "node --test ./javascript/test.mjs",
+    "bench": "node ./javascript/bench.js"
+  },
+  "main": "javascript/dist/cjs/simsimd.js",
+  "module": "javascript/dist/esm/simsimd.js",
+  "exports": {
+    ".": {
+      "import": "./javascript/dist/esm/simsimd.js",
+      "require": "./javascript/dist/cjs/simsimd.js"
+    }
+  },
+  "engines": {
+    "node": "~10 >=10.20 || >=12.17"
+  },
+  "dependencies": {
+    "bindings": "^1.5.0",
+    "node-addon-api": "^3.2.1",
+    "node-gyp-build": "^4.8.0"
+  },
+  "devDependencies": {
+    "@types/bindings": "^1.5.5",
+    "@types/node": "^20.12.2",
+    "node-gyp": "^10.0.1",
+    "prebuildify": "^6.0.0",
+    "typescript": "^5.3.3"
+  },
+  "optionalDependencies": {
+    "benchmark": "^2.1.4",
+    "mathjs": "^11.11.2",
+    "usearch": "^2.8.15"
+  }
+}

package/simsimd/rust/benches/cosine.rs

@@ -0,0 +1,27 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
+use simsimd::SpatialSimilarity as SimSIMD;
+
+mod native;
+
+const DIMENSIONS: usize = 1536;
+
+pub fn cos_benchmark(c: &mut Criterion) {
+    let inputs: (Vec<f32>, Vec<f32>) = (
+        native::generate_random_vector(DIMENSIONS),
+        native::generate_random_vector(DIMENSIONS),
+    );
+
+    let mut group = c.benchmark_group("SIMD Cosine");
+
+    for i in 0..=5 {
+        group.bench_with_input(BenchmarkId::new("SimSIMD", i), &i, |b, _| {
+            b.iter(|| SimSIMD::cosine(&inputs.0, &inputs.1))
+        });
+        group.bench_with_input(BenchmarkId::new("Rust Native", i), &i, |b, _| {
+            b.iter(|| native::cos_similarity_cpu(&inputs.0, &inputs.1))
+        });
+    }
+}
+
+criterion_group!(benches, cos_benchmark);
+criterion_main!(benches);

package/simsimd/rust/benches/native.rs

@@ -0,0 +1,30 @@
+#![allow(unused)]
+use rand::Rng;
+
+pub(crate) fn generate_random_vector(dim: usize) -> Vec<f32> {
+    (0..dim).map(|_| rand::thread_rng().gen()).collect()
+}
+
+pub(crate) fn cos_similarity_cpu(a: &[f32], b: &[f32]) -> Option<f32> {
+    if a.len() != b.len() {
+        return None;
+    }
+
+    let (dot_product, norm_a, norm_b) = a
+        .iter()
+        .zip(b)
+        .map(|(a, b)| (a * b, a * a, b * b))
+        .fold((0.0, 0.0, 0.0), |acc, x| {
+            (acc.0 + x.0, acc.1 + x.1, acc.2 + x.2)
+        });
+
+    Some(1.0 - (dot_product / (norm_a.sqrt() * norm_b.sqrt())))
+}
+
+pub(crate) fn squared_euclidean_cpu(a: &[f32], b: &[f32]) -> Option<f32> {
+    if a.len() != b.len() {
+        return None;
+    }
+
+    Some(a.iter().zip(b).map(|(x, y)| (x - y).powi(2)).sum())
+}

package/simsimd/rust/benches/sqeuclidean.rs

@@ -0,0 +1,27 @@
+use criterion::{criterion_group, criterion_main, BenchmarkId, Criterion};
+use simsimd::SpatialSimilarity as SimSIMD;
+
+mod native;
+
+const DIMENSIONS: usize = 1536;
+
+pub fn l2sq_benchmark(c: &mut Criterion) {
+    let inputs: (Vec<f32>, Vec<f32>) = (
+        native::generate_random_vector(DIMENSIONS),
+        native::generate_random_vector(DIMENSIONS),
+    );
+
+    let mut group = c.benchmark_group("SIMD SqEuclidean");
+
+    for i in 0..=5 {
+        group.bench_with_input(BenchmarkId::new("SimSIMD", i), &i, |b, _| {
+            b.iter(|| SimSIMD::sqeuclidean(&inputs.0, &inputs.1))
+        });
+        group.bench_with_input(BenchmarkId::new("Rust Native", i), &i, |b, _| {
+            b.iter(|| native::squared_euclidean_cpu(&inputs.0, &inputs.1))
+        });
+    }
+}
+
+criterion_group!(benches, l2sq_benchmark);
+criterion_main!(benches);