Skip to content

add rng option to jittablemodule#85

Open
qihqi wants to merge 2 commits intogoogle:mainfrom
qihqi:codex/add-rng-option-to-jittablemodule
Open

add rng option to jittablemodule#85
qihqi wants to merge 2 commits intogoogle:mainfrom
qihqi:codex/add-rng-option-to-jittablemodule

Conversation

@qihqi
Copy link
Copy Markdown
Collaborator

@qihqi qihqi commented Mar 31, 2026

Add an option for JittableModule to take in rng. This Pr addresses the issue in #17 where models with dropout want to use different rng behavior every run.

@qihqi qihqi requested review from QiliangCui and wdhongtw March 31, 2026 00:34
@wdhongtw
Copy link
Copy Markdown
Collaborator

Thank for the PR. I've noted the changes.
I need a few business days for a thorough review. Will get back to it soon.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants