Vertex embeddings module #1435
Draft
Vertex embeddings module #1435
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
| The latest updates on your projects. Learn more about Vercel for GitHub.
|
mattkjames7 added the feature 18 tasks
github-merge-queue bot pushed a commit to memgraph/mage that referenced this pull request Oct 15, 2025
The issue is that during Dockerfile build, CPU version gets installed, when memgraph starts, it loads the CPU version, and then it's not possible to reload the loaded module using the `CALL mg.load_all()`. Should we only install the GPU version -> https://stackoverflow.com/questions/58511598/can-both-the-gpu-and-cpu-versions-of-pytorch-be-installed-in-the-same-conda-envi? -> NO, because of the size, take a look at the comment below ### Description Added: - `embeddings` module to compute vertex sentence embeddings. - CUDA-enabled build of MAGE for daily and RC builds. ### Pull request type - [ ] Bugfix - [x] Algorithm/Module - [ ] Feature - [ ] Code style update (formatting, renaming) - [ ] Refactoring (no functional changes, no api changes) - [ ] Build related changes - [ ] Documentation content changes - [ ] Other (please describe): ###################################### ### Reviewer checklist (the reviewer checks this part) #### Module/Algorithm - [ ] Core algorithm/module implementation - [ ] [Query module](https://memgraph.com/docs/memgraph/reference-guide/query-modules) implementation - [ ] Tests provided (unit / e2e) - [ ] Code documentation - [ ] README short description ### Documentation checklist - [x] Add the documentation label tag - [x] Add the bug / feature label tag - [x] Add the milestone for which this feature is intended - If not known, set for a later milestone - [x] Write a release note, including added/changed clauses - Added `embedding` module, allowing users to compute sentence embeddings for vertices on CPU or Nvidia GPU. [#662](#662). - [x] Link the documentation PR here - memgraph/documentation#1435 --------- Co-authored-by: Dr Matt James <mattkjames7@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit. This suggestion is invalid because no changes were made to the code. Suggestions cannot be applied while the pull request is closed. Suggestions cannot be applied while viewing a subset of changes. Only one suggestion per line can be applied in a batch. Add this suggestion to a batch that can be applied as a single commit. Applying suggestions on deleted lines is not supported. You must change the existing code in this line in order to create a valid suggestion. Outdated suggestions cannot be applied. This suggestion has been applied or marked resolved. Suggestions cannot be applied from pending reviews. Suggestions cannot be applied on multi-line comments. Suggestions cannot be applied while the pull request is queued to merge. Suggestion cannot be applied right now. Please check back later.
Release note
Added
embeddingmodule, allowing users to compute sentence embeddings for vertices on CPU or Nvidia GPU. memgraph/mage#662.Related product PRs
PRs from product repo this doc page is related to:
memgraph/mage#662
Checklist:
bugfixorfeaturelabel, based on the product PR type you're documenting