jax

34c5163f - Refactored common upcast for integral-type accumulators