``scan`` transformations for brainpy `Variable` system
chaoming0625 opened this issue · 1 comments
chaoming0625 commented
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.
chaoming0625 commented
Has been supported.