How to convert back from brainpy arrays to functional-style jax

[llandsmeer]

BrainPy uses a mutable/OOP style interface and has converters from functional jax style to BrainPy arrays. I'm using a project with functional jax, and would like to use brainpy inside this project. Thus, I would like to use a brainpy model in a functional style again, ie extract the function

timestep: state, input -> state'

How do I do this?

[chaoming0625]

I think you should use brainstate. It supports the functionality of inside of the jit function.

[github-actions[bot]]

This issue has been automatically marked as stale because it has not had recent activity in the last 90 days. It will be closed in 14 days if no further activity occurs. If this issue is still relevant, please comment to keep it open. Thank you for your contributions!