DOC: Improved docs for polynomial arithmetic functions of jax.numpy

[rajasekharporeddy]

This PR includes better docs and example code snippets for jnp.polyadd, jnp.polysub, jnp.polymul and jnp.polydiv functions.

Part of #21461

[rajasekharporeddy]

Hi @jakevdp, Please suggest if I need to make any other changes.

[jakevdp]

Please remove the unused import & fix conflicts!

Also, please comment when you've done this because I don't get notifications if you just push the change (that's why I'm so late coming back to this)

[rajasekharporeddy]

Thanks! Removed the unused import and resolved the conflicts. Also, squashed into single commit.

[rajasekharporeddy]

Hi @jakevdp, Could you please have a look at this PR.

[dfm]

@rajasekharporeddy — Jake is offline for the next little bit and I'm looking after these docstring update PRs. I'm sorry for missing this yesterday, and please feel free to ping me if I'm missing anything else!

This all looks good - thanks!

[rajasekharporeddy]

Thanks @dfm! Fixed the traceback in examples and squashed into single commit.

[dfm]

@rajasekharporeddy — I believe the that traceback lines still need to be updated. They should start with:

Traceback (most recent call last):

instead of

Traceback: most recent call last

Note the parentheses and colon location. Thanks!

[rajasekharporeddy]

Modified the traceback as suggested.

[dfm]

Looks good - thanks!