Enable brainpy object as pytree so that it can be applied with jax.jit etc. directly
#876
The logs for this run have expired and are no longer available.
Loading
jax.jit etc. directly
#876