Skip to content

Releases: Achuttarsing/umap-gpu

v0.3.0

Choose a tag to compare

@Achuttarsing Achuttarsing released this 12 Jun 15:47

Align the UMAP implementation with the reference pure-MLX UMAP so it produces equivalent embeddings.

Changes

  • Spectral initialization (new src/spectral-init.ts): power iteration on the symmetric-normalized graph adjacency, scaled to [0,10] + jitter — replaces random uniform init.
  • smooth-kNN σ search: sums over neighbors j=1..k-1 (skips the j=0/rho column), rho = max(min_nonzero, 1e-8), σ bounded to [1e-20, 1e3] then floored at 1e-10.
  • SGD: squared distances floored at 1e-6; each active edge does exactly negativeSampleRate repulsions (flat rate); the CPU path now uses Jacobi accumulation to match the GPU two-pass.
  • Made the CPU-vs-GPU comparison test deterministic.

Published automatically via the GitHub Actions npm pipeline (OIDC trusted publishing).

v0.2.16

Choose a tag to compare

@Achuttarsing Achuttarsing released this 08 Apr 11:22
01a4677

What's new

  • Add seed option for deterministic output — pass seed: number to UMAPOptions to get reproducible embeddings across runs. When omitted, behavior is unchanged.

v0.2.15

Choose a tag to compare

@Achuttarsing Achuttarsing released this 02 Mar 09:55
dc59664

What's Changed

Full Changelog: v0.2.14...v0.2.15

v0.2.14

Choose a tag to compare

@Achuttarsing Achuttarsing released this 01 Mar 21:44
17e6527

What's Changed

  • Claude/fix umap critical bugs

Full Changelog: v0.2.13...v0.2.14

v0.2.13

Choose a tag to compare

@Achuttarsing Achuttarsing released this 01 Mar 20:43
b1ae631

What's Changed

Full Changelog: v0.2.12...v0.2.13

v0.2.12

Choose a tag to compare

@Achuttarsing Achuttarsing released this 01 Mar 20:38
d5a174a

What's Changed

  • fix: correct epochsPerSample formula and nNeg calculation in CPU SGD by @Achuttarsing in #20

Full Changelog: v0.2.11...v0.2.12

v0.2.11

Choose a tag to compare

@Achuttarsing Achuttarsing released this 28 Feb 23:51
b0fdb07

What's Changed

  • Fix kNN distance smoothing and SGD gradient application by @Achuttarsing in #19

Full Changelog: v0.2.10...v0.2.11

v0.2.10

Choose a tag to compare

@Achuttarsing Achuttarsing released this 28 Feb 23:37
ccc4d8c

What's Changed

  • Fix symmetrization and attraction gradient in SGD layout by @Achuttarsing in #18

Full Changelog: v0.2.9...v0.2.10

v0.2.9

Choose a tag to compare

@Achuttarsing Achuttarsing released this 28 Feb 23:06
a1d84f4

What's Changed

  • Add docs deployment workflow and update API documentation by @Achuttarsing in #16
  • Improve package.json metadata and export configuration by @Achuttarsing in #17

Full Changelog: v0.2.8...v0.2.9

v0.2.8

Choose a tag to compare

@Achuttarsing Achuttarsing released this 28 Feb 21:33
56e2b0a

What's Changed

Full Changelog: v0.2.7...v0.2.8