[FEATURE] Improve type annotation (and documentation?) for `TriangleMesh.at` indexing methods

[jeertmans]

Terms

• Checked the existing issues to see if my suggestion has not already been suggested;

Description

Currently, the TriangleMesh.at methods, such as add, do not provide information about the accepted keyword arguments.

I'm not sure if it's feasible or worth the hassle, but wouldn't it be nice to automatically derive allowed keyword arguments (and maybe their documentation) from jax.numpy.ndarray.at.get?

Screenshots

Additional information

https://docs.jax.dev/en/latest/_autosummary/jax.numpy.ndarray.at.html