4.0.2 Optional JAX build

What's Changed

• Do not use removed numpy.bool by @mattwthompson in #485
• switch to using mamba for rtd building by @mikemhenry in #494
• Minor docstring exp fix by @ijpulidos in #492
• Make JAX an optional dependency by @Lnaden in #503
• Warn about JAX bitsize changes by @Lnaden in #504
• Correctly stagger JIT until first call by @Lnaden in #505
• Correct docstring default value by @badisa in #507
• Wrap the preconditioned jit by @Lnaden in #506
• fixes RTD build by @mikemhenry in #510
• Fix a small bug in the timeseries test, test against min and max versions by @Lnaden in #512

New Contributors

• @mattwthompson made their first contribution in #485
• @ijpulidos made their first contribution in #492
• @badisa made their first contribution in #507

Full Changelog: 4.0.1...4.0.2