jax

c2f64619 - Merge pull request #40079 from jakevdp:jax-export-typing