source

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.


source

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.


source

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.


source

TraitCollection

 TraitCollection (loci_loc:jax.Array, add_eff:jax.Array,
                  intercept:jax.Array)

A collection of additive genetic traits.