[JAX] Update the ANN document.
[JAX] Update the ANN document.

Deletes the documentation that explains the algorithm.
I don't think it is the necessary detail for users.
We'll write a paper to explain it in detail very soon.

PiperOrigin-RevId: 426546480

google/jax,google/jax,google/jax,google/jax