All modules for which code is available

• cellarium.ml.callbacks.compute_norm
• cellarium.ml.callbacks.get_coord_data
• cellarium.ml.callbacks.loss_scale_monitor
• cellarium.ml.callbacks.prediction_writer
• cellarium.ml.callbacks.variance_monitor
• cellarium.ml.cli
• cellarium.ml.core.datamodule
• cellarium.ml.core.module
• cellarium.ml.core.pipeline
• cellarium.ml.data.dadc_dataset
• cellarium.ml.data.distributed_anndata
• cellarium.ml.data.fileio
• cellarium.ml.data.pytree_dataset
• cellarium.ml.data.schema
• cellarium.ml.distributions.negative_binomial
• cellarium.ml.layers.attention
• cellarium.ml.layers.embedding
• cellarium.ml.layers.ffn
• cellarium.ml.layers.head
• cellarium.ml.layers.mu_linear
• cellarium.ml.layers.normadd
• cellarium.ml.layers.transformer
• cellarium.ml.lr_schedulers.linear_lr
• cellarium.ml.models.cellarium_gpt
• cellarium.ml.models.geneformer
• cellarium.ml.models.incremental_pca
• cellarium.ml.models.logistic_regression
• cellarium.ml.models.model
• cellarium.ml.models.onepass_mean_var_std
• cellarium.ml.models.probabilistic_pca
• cellarium.ml.models.tdigest
• cellarium.ml.preprocessing.highly_variable_genes
• cellarium.ml.transforms.cellarium_gpt_tokenizer
• cellarium.ml.transforms.divide_by_scale
• cellarium.ml.transforms.filter
• cellarium.ml.transforms.log1p
• cellarium.ml.transforms.normalize_total
• cellarium.ml.transforms.z_score
• cellarium.ml.utilities.data
• cellarium.ml.utilities.distributed
• cellarium.ml.utilities.testing