Update setup.py
I was tweaking m4's `setup.py` with some goodies from here and saw a few minor improvements that could be made here:

- clarify the license and description
- add `jax` to the keywords
- add recent python versions

If anything doesn't fit please let me know and I will remove it.
_The documentation is not available anymore as the PR was closed or merged._