Initialization

src.jaxnasium.algorithms.utils.rl_initialization(key: PRNGKeyArray, network: eqx.Module, weight_init: jax.nn.initializers.Initializer = jax.nn.initializers.orthogonal(), bias_init=0.0)

Sets all layers in a network to a given weight and bias initialization. Defaults to orthogonal weight initialization and zero bias initialization.