Add way to set backend `fn` random generators#7629

Open

lucianopaz wants to merge 9 commits intopymc-devs:mainfrom

lucianopaz:trace_rng

Commits on Jan 10, 2025

Set rng state for trace fn mapping draws to posterior samples

lucianopaz
and
ricardoV94
committed
copy_function_with_new_rngs warns with JAXLinker
lucianopaz
committed