https://github.com/apple/embedding-atlas

Embedding Atlas is a tool that provides interactive visualizations for large embeddings. It allows you to visualize, cross-filter, and search embeddings and metadata.

Features

๐Ÿท๏ธ Automatic data clustering & labeling: Interactively visualize and navigate overall data structure.

๐Ÿซง Kernel density estimation & density contours: Easily explore and distinguish between dense regions of data and outliers.

๐ŸงŠ Order-independent transparency: Ensure clear, accurate rendering of overlapping points.

๐Ÿ” Real-time search & nearest neighbors: Find similar data to a given query or existing data point.

๐Ÿš€ WebGPU implementation (with WebGL 2 fallback): Fast, smooth performance (up to few million points) with modern rendering stack.

๐Ÿ“Š Multi-coordinated views for metadata exploration: Interactively link and filter data across metadata columns.

Please visit https://apple.github.io/embedding-atlas for a demo and documentation.