jnp_to_np ========= .. currentmodule:: energnn.graph .. autofunction:: jnp_to_np