brainstate.interop.from_equinox

brainstate.interop.from_equinox(model, *, sample_input=None)

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

Parameters:

• model (Any) – The source model. Either a single registered layer or an eqx.nn.Sequential.

• sample_input (Any) – Required when the model contains convolution layers.

Returns:

The converted model.

Return type:

Any