Should self.update_states(), self.compute_current(), and self.init_states() be staticmethod?

[michaeldeistler]

Currently, all of the above methods are not typically defined to be static. This can lead users to think that they can just use self.channel_params to obtain the parameters in def update_states().

To avoid this misuse, we could just mark all of these methods as static (at least in the tutorials) and thereby allow them to not have any global state.

[michaeldeistler]

Another option would be to just del channel.channel_params after they have been copied over to self.nodes

[jnsbck]

I prefer the 2nd option, since they are already contained in nodes.

[jnsbck]

init_states cannot be jitted since it acts on nodes. @michaeldeistler and I discussed changing it to sth. like.

cell.to_jax()
cell.init_states()
v = jx.integrate(cell)

this would allow:

cell.to_jax()

@jax.jit
def simulate():
    cell.init_states()
    v = jx.integrate(cell)
    return v