brainstate.interop.to_equinox

brainstate.interop.to_equinox(model, *, key=None)

Convert a brainstate.nn model into an equivalent equinox model.

Parameters:

• model (Any) – The source model.

• key (Any) – Key used to construct the equinox layers (weights are overwritten immediately).

Returns:

The converted model.

Return type:

Any