Update transform to use np for dk, not jnp
Update transform to use np for dk, not jnp

-using jnp broke some of the equals stuff when comparing dk array
-it's precomputed so jnp not needed
-just using regular np now

PlasmaControl/DESC,PlasmaControl/DESC