Upload 2 files

jax_params/config.json

@@ -0,0 +1 @@
+{"n_genes": 19062, "n_expressions_bins": 64, "embed_dim": 256, "init_gene_embed_dim": 200, "project_gene_embedding": true, "use_gene_embedding": true, "num_attention_heads": 8, "key_size": 32, "ffn_embed_dim": 512, "num_layers": 4, "use_memory_efficient_attention": false, "use_gradient_checkpointing": true, "gene2vec_weights_path": "data/gene2vec_weights_common_gene_ids.npy", "embeddings_layers_to_save": [], "attention_layers_to_save": [], "use_log_normalization": true, "use_max_normalization": true, "normalization_factor": 5.547176906585117}

jax_params/params.joblib

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