brainstate.interop.to_linen# brainstate.interop.to_linen(model)[source]# Convert a brainstate.nn model into a flax.linen (module, params) pair. Parameters: model (Any) – The source model. Returns: (linen_module, params) where params is a FrozenDict accepted by linen_module.apply. Return type: Tuple[Any, Any]
