Upload 2 files

jax_params/config.json

@@ -0,0 +1 @@
+{"name": "MOJO", "alphabet_size": {"rnaseq": 66, "methylation": 66}, "token_embed_dim": 256, "conv_init_embed_dim": 512, "embed_dim": 512, "num_downsamples": 8, "filter_list": [512, 512, 512, 512, 512, 512, 512, 512, 512], "init_gene_embed_dim": 200, "project_gene_embedding": true, "use_gene_embedding": true, "sequence_length": 17116, "fixed_sequence_length": 17152, "use_remat_in_convs": false, "use_remat_in_transformer": true, "use_skip_connection": true, "embeddings_layers_to_save": [], "attention_layers_to_save": [], "gene2vec_weights_path": "data/gene2vec_weights_methylation.npy", "num_attention_heads": 16, "key_size": 32, "ffn_embed_dim": 1024, "num_layers": 8, "layer_norm_eps": 1e-05, "stem_kernel_shape": 15, "num_hidden_layers_head": 1, "n_expressions_bins": {"rnaseq": 64, "methylation": 64}, "min_omic_value": {"rnaseq": 0.0, "methylation": 0.006863548701544}, "max_omic_value": {"rnaseq": 1.0, "methylation": 0.992831582796998}, "use_log_normalization": {"rnaseq": true, "methylation": false}, "use_max_normalization": {"rnaseq": true, "methylation": false}, "normalization_factor": {"rnaseq": 5.52786861525666, "methylation": 1.0}}

jax_params/params.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7181b903fe91eea5af6e79a7fb3e1bf69aba9fd936c98e5f02866b364e7f9531
+size 209201151