SplitRngs ========= Splits RNG keys for stochastic operations. **Abstract Signature:** ``SplitRngs(key, count: int = 2)`` .. raw:: html

JAX (Core)

API: jax.random.split
Strategy: Direct Mapping

Flax NNX

API: flax.nnx.split_rngs
Strategy: Direct Mapping