Rngs

A small abstraction to manage RNG state.

Abstract Signature:

Rngs(seed: int = 0)

PyTorch

API: torch.Generator
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: flax.nnx.Rngs
Strategy: Direct Mapping
Official Docs ↗