### Add your issue here Discussed in #999, we are going to have two `iternorm` implementations. One for JAX and one for everything else.
Add your issue here
Discussed in #999, we are going to have two
iternormimplementations. One for JAX and one for everything else.