Here are the documentation pages for each library in the JAX scientific ecosystem:

Here are the documentation pages for other libraries:

Many of my smaller libraries have their documentation as part of their GitHub README. A full list of such libraries is available here: https://github.com/patrick-kidger