traveler.tables.JaxTable.with_prng_keys#

JaxTable.with_prng_keys(id_col, *, seed, namespace, force=False)[source]#

Return a table with deterministic per-row PRNG keys materialized.

Existing keys are reused when their configuration matches. Passing a different seed, namespace, or ID column creates a new key column.

Parameters:

  • id_col (str)

  • seed (int)

  • namespace (str)

  • force (bool)

Return type:

Self