Add JAX-based differentiable RF optimization example for komamripy

[anvika-singhal]

Feature Request

Provide an end-to-end differentiable optimization example using JAX. Finalize JAX integration approach.
Implement example:

• Call simulation from Python
• Compute gradients
• Optimize RF pulse