DOC: add a table of contents for top level API docs (#2946)
DOC: add a table of contents for top level API docs (#2946)

This makes them easier to scan.
google/jax,google/jax,google/jax,google/jax,tensorflow/probability,tensorflow/probability