GH-44229: [Docs] Add PyArrow to JAX example to the docs

GH-44229: [Docs] Add PyArrow to JAX example to the docs #77506