Comment by schizo89

Neural differential equations are also easier with jax. sim2real may be easier with simulator where some of hard computations are replaced with neural approximations