synax.GRU.init_params

synax.GRU.init_params(key: Key) → dict[str, jax.Array]

Sample initial parameters.

Parameters:

key – PRNG key.

Returns:

Parameters.