[DOC] We should document all jax config from the file config.py

[nouiz]

Please:

• Check for duplicate requests.
• Describe your goal, and if possible provide a code snippet with a motivating example.

We should document all jax config from the file config.py
We can search the file config.py, but that isn't a great user experience. Most user look in docs first, only few of them look at the source code.