Enable brainpy object as pytree so that it can be applied with jax.jit etc. directly#625

chaoming0625 merged 2 commits intomasterbrainpy/BrainPy:masterfrom

brainpy_object_as_pytreebrainpy/BrainPy:brainpy_object_as_pytreeCopy head branch name to clipboard