jax

909fee6a - try adding sphinx-autodoc-typehints