brainpy/BrainPy

``scan`` transformations for brainpy `Variable` system

chaoming0625 opened this issue · 1 comments

Please:

  • Check for duplicate requests.
  • Describe your goal, and if possible provide a code snippet with a motivating example.

Although there is brainpy.math.for_loop, it does not support the loop over the non-Variable instances. Instead, jax.lax.scan supports iteratively changing the value over inputs.

Has been supported.