We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
9a2245d
hk.mixed_precision.push_policy
hk.experimental.{get_params,get_initial_state,get_current_state}
hk.experimental.{maybe_get_rng_sequence_state,maybe_replace_rng_sequence_state}
hk.switch
hk.get_parameter
init=None
hk.MethodContext
orig_class
hk.GetterContext
lifted_prefix_name
hk.layer_stack
jax.enable_custom_prng
TruncatedNormal
dict
Mapping
hk.dropout
broadcast_dims