-
Notifications
You must be signed in to change notification settings - Fork 24.2k
Add optional generator to distribution sampler/rsample methods. #146333
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: main
Are you sure you want to change the base?
Add optional generator to distribution sampler/rsample methods. #146333
Conversation
The committers listed above are authorized under a signed CLA. |
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/146333
Note: Links to docs will display an error until the docs builds have been completed. ✅ No FailuresAs of commit fd5c9aa with merge base 06850e6 ( This comment was automatically generated by Dr. CI and updates every 15 minutes. |
4586e2d
to
fd5c9aa
Compare
@pytorchbot label "module: distributions" |
@pytorchbot label "topic: not user facing" |
Are we able to move this PR along by chance? Would be helpful for our work in pytensor and unifying the API. |
Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as |
Fixes part of #45115 and #11340
Adds a generator parameter to all the sample/rsample methods of torch distribution classes
cc @fritzo @neerajprad @alicanb @nikitaved