source

Population

 Population (geno:jax.Array, ibd:jax.Array, id:jax.Array, iid:jax.Array,
             mother:jax.Array, father:jax.Array, sex:jax.Array,
             gen:jax.Array, pheno:jax.Array, fixEff:jax.Array,
             is_active:jax.Array, gv:Optional[jax.Array]=None,
             bv:Optional[jax.Array]=None, dd:Optional[jax.Array]=None,
             aa:Optional[jax.Array]=None, ebv:Optional[jax.Array]=None,
             misc:Optional[Dict[str,Any]]=None,
             miscPop:Optional[Dict[str,Any]]=None)

User-facing container for population data (setup & analysis). The JIT carry uses SimState (internal) instead.


source

SimParam

 SimParam (gen_map:jax.Array, ploidy:int, traits:Optional[Any]=None,
           recomb_params:tuple=(2.6, 0.0, 0.0),
           var_e:Optional[jax.Array]=None)

Global, run-constant parameters & metadata. Keep only JAX-compatible fields (arrays, tuples, scalars).