[ge_arrow] Update to JAX

[xuanguang-li]

Updated the ge_arrow.md to JAX and complemented the styling consistent with the operation manual.

Key changes:

• Rewrite the RecurCompetitive class as a NamedTuple.
• Complete all computations inside the compute_rc_model function. Inside this function, arguments of sub-functions can be written in the same way as the definitions in the theory part.
• Partially jitted the main computation function, and used jax.lax.fori_loop to conduct loops.
• Fixed some typos and styling.