Shypd.aiThe largest open index of AI toolsPyMDE is a Python library for computing vector embeddings using minimum-distortion embedding techniques with PyTorch. It provides a general framework for creating and recreating various embeddings, optimized for speed on both CPU and GPU.
Trending
About
PyMDE is a Python library designed for computing vector embeddings for finite sets of items, such as images, biological cells, or network nodes. Built with PyTorch, it offers a simple yet general framework called Minimum-Distortion Embedding (MDE), allowing users to easily recreate well-known embeddings or develop new ones tailored to specific applications. PyMDE is competitive in runtime with more specialized embedding methods, with even faster performance on a GPU. It features fast preprocessing routines implemented in Rust, including approximate and exact k-nearest neighbor algorithms and breadth-first search for all-pairs shortest paths. PyMDE can be used to visualize datasets, generate feature vectors for supervised learning, compress high-dimensional data, and draw graphs efficiently.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
