Jax: Fast Combinations Calculation
Combinadics A fast combinations calculation in jax. Idea of combinadic implementation is from https://jamesmccaffrey.wordpress.com/2022/06/28/generating-the-mth-lexicographical-element-of-a-combination-using-the-combinadic and some useful information can be found here: https://en.wikipedia.org/wiki/Combinatorial_number_system. Below I copied and aggregated some of the details. Introduction The following code demostrates the combinations calculation in numpy and via combinadics: # setup n =