Model.set_seed()#
- Model.set_seed(seed)[source]#
Splits and sets the seed / PRNG key.
- Parameters:
seed (
PRNGKeyArray
) – The seed is split and distributed to the seed nodes of the model. Must be aKeyArray
, i.e. an array of shape (2,) and dtypeuint32
. Seejax.random
for more details.- Return type: