docs: Update JAX backend documentation on training support#5482
Merged
background
wait
wait-all
cancel
parallel
Loading