Jax: Fast Combinations Calculation