add_trait_a
add_trait_a (key:<function PRNGKey>, founder_pop:chewc.structs.Population, sim_param:chewc.structs.SimParam, n_qtl_per_chr:int, mean:jax.Array, var:jax.Array, cor_a:Optional[jax.Array]=None, gamma:bool=False, shape:float=1.0)
Adds one or more new additive traits to the simulation parameters. This function is run on the host, so boolean indexing is fine here.
set_pheno
set_pheno (key:<function PRNGKey>, pop:chewc.structs.Population, sp:chewc.structs.SimParam, var_e:jax.Array)
Computes and sets phenotypes with a fixed environmental variance.
set_pheno_h2
set_pheno_h2 (key:<function PRNGKey>, pop:chewc.structs.Population, sp:chewc.structs.SimParam, h2:Union[float,jax.Array], broad_sense:bool=False)
Computes and sets phenotypes based on a target heritability. This function is now fully JIT-compatible.
TraitCollection
TraitCollection (loci_loc:jax.Array, add_eff:jax.Array, intercept:jax.Array)
A collection of additive genetic traits.