Thread PRNG key through wrap_jax_jit to prevent UnexpectedTracerError#95
Open
ifding wants to merge 1 commit into
Open
Thread PRNG key through wrap_jax_jit to prevent UnexpectedTracerError#95ifding wants to merge 1 commit into
ifding wants to merge 1 commit into
