* add gumbel sampling#94
Open
mattlST wants to merge 3 commits into
Open
Conversation
added 3 commits
March 24, 2026 22:11
- gumbel wrapper takes right now only RWKV, needs an autoregressive network with stepwise sampling implementation - RWKV implementation based on Jonas Rigo - some basis tests for gumbel sampling - adapted sampler and vqs to support gumbel sampling
- a comment about implementing setting MPI rank to 0 in gumbel_test (but probably deprecated soon)
to identify which lDim is being tested
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
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
"Manually rebased" on master
Let's see if the tests work now, locally (cpu) all tests passed
Old pull request:
Hi Markus,
mostly add a sample technique without repetition, aka gumbel sampling.
The implementation is based on the article: https://arxiv.org/pdf/2408.07625v1
The algorithm is implemented for general local Hilbert space dimension, I mostly used it for peaked bosonic systems (Bose Hubbard).
The core addition is the gumbel_wrapper in the util subfolder, some remarks about the wrapper:
I hope this can be useful to some folks! Most likely, I forgot something important, so if there are any questions, problems, let me know!
Best,
Mathias