synax.Parallel.init

synax.Parallel.init(key: Key) tuple[Any, Ellipsis][source]

Sample initial parameters.

Parameters:

key – PRNG key.

Returns:

Parameters.