rnglib

Contents

rnglib#

class flax.experimental.nnx.Rngs(*args, **kwargs)[source]#
class flax.experimental.nnx.RngStream(*args: 'Any', **kwargs: 'Any')[source]#