jax

2dc81fb4 - Make pytest run over JAX tests warning clean, and error on warnings. (#2674)

Make pytest run over JAX tests warning clean, and error on warnings. (#2674) * Make pytest run over JAX tests warning clean, and error on warnings. Remove global warning suppression in travis.yml. Instead add a pytest.ini that converts warnings to errors, with the exception of a whitelist. Either fix or locally suppress warnings in tests. Also fix crashes on Mac related to a preexisting linear algebra bug. * Fix some type errors in the FFT transpose rules revealed by the convert_element_type transpose rule change.