[nbdev]
jaxDiversity = jaxDiversity._modidx:d
