jax

216bd9a6 - Fix dtype issue in stirling approximation