FromFlax

class brainpy.dnn.FromFlax(flax_module, *module_args, **module_kwargs)

Transform a Flax module as a BrainPy DynamicalSystem.

Parameters:

• flax_module (Any) – The flax Module.

• module_args (Any) – The module arguments, used to initialize model parameters.

• module_kwargs (Any) – The module arguments, used to initialize model parameters.

update(*args, **kwargs)

The function to specify the updating rule.