.dxtignore
.gitattributes
.gitignore
.nojekyll
.pre-commit-config.yaml
CITATION.cff
CLAUDE.md
LICENSE
README.md
env.yml
icon.png
index.html
manifest.json
pyproject.toml
pytest.ini
requirements.txt
uv.lock
.claude/agents/code-reviewer.md
.claude/agents/test-suite-architect.md
.cursor/rules/linting.mdc
.cursor/rules/test-driven-dev.mdc
.cursor/rules/uv-run-rule.mdc
.github/pull_request_template.md
.github/workflows/publish-pypi.yml
.github/workflows/push.yml
assets/lobster.png
docs/CONTRIBUTING.md
docs/CONTRIBUTORS.md
docs/DXT_DISTRIBUTION.md
docs/MCP_INTEGRATION.md
docs/RL_TRAINING.md
docs/UME_GRPO_LOGGING.md
docs/diagrams/create_ume_architecture_diagram.py
docs/diagrams/ume_architecture_diagram.html
examples/compare_flash_attention_embeddings.py
examples/example_checkpoint_compatibility.py
examples/example_embed_fa2.py
examples/example_sdpa_unpadded_attention_mask.py
examples/generate_synthetic_dataset.py
examples/inference.py
examples/intervene.py
examples/onnx_example.py
examples/perturbation_score_example.py
examples/train_ume_grpo.py
examples/ume_hf_example.py
examples/ume_hf_sanity_check.py
notebooks/01-inference.ipynb
notebooks/02-intervention.ipynb
notebooks/03-architecture-analyzer.ipynb
notebooks/04-ume-multimodal-embeddings.ipynb
scripts/check_claude_md.py
slurm/README.md
slurm/scripts/eval_dgeb_mock_baseline.sh
slurm/scripts/eval_dgeb_ume.sh
slurm/scripts/evaluate_ume.sh
slurm/scripts/train_neobert.sh
slurm/scripts/train_ume-2.sh
slurm/scripts/train_ume.sh
slurm/scripts/train_ume_trl.sh
src/lbster.egg-info/PKG-INFO
src/lbster.egg-info/SOURCES.txt
src/lbster.egg-info/dependency_links.txt
src/lbster.egg-info/entry_points.txt
src/lbster.egg-info/requires.txt
src/lbster.egg-info/top_level.txt
src/lobster/__init__.py
src/lobster/_ensure_package.py
src/lobster/assets/__init__.py
src/lobster/assets/ab_vocab.txt
src/lobster/assets/pmlm_vocab.txt
src/lobster/assets/protein_vocab.txt
src/lobster/assets/3di_tokenizer/__init__.py
src/lobster/assets/3di_tokenizer/special_tokens_map.json
src/lobster/assets/3di_tokenizer/tokenizer_config.json
src/lobster/assets/3di_tokenizer/vocab.txt
src/lobster/assets/amino_acid_tokenizer/__init__.py
src/lobster/assets/amino_acid_tokenizer/special_tokens_map.json
src/lobster/assets/amino_acid_tokenizer/tokenizer.json
src/lobster/assets/amino_acid_tokenizer/tokenizer_config.json
src/lobster/assets/cdna_tokenizer/__init__.py
src/lobster/assets/cdna_tokenizer/special_tokens_map.json
src/lobster/assets/cdna_tokenizer/tokenizer_config.json
src/lobster/assets/cdna_tokenizer/vocab.txt
src/lobster/assets/codon_tables/codon_table.json
src/lobster/assets/codon_tables/vendor_codon_table.json
src/lobster/assets/concepts/biological_process_unique_values.txt
src/lobster/assets/concepts/cellular_component_unique_values.txt
src/lobster/assets/concepts/cluster_name_unique_values.txt
src/lobster/assets/concepts/members_unique_values.txt
src/lobster/assets/concepts/molecular_function_unique_values.txt
src/lobster/assets/concepts/organism_unique_values.txt
src/lobster/assets/concepts/rep_id_unique_values.txt
src/lobster/assets/concepts/taxon_id_unique_values.txt
src/lobster/assets/concepts/taxon_unique_values.txt
src/lobster/assets/concepts/unique_identifier_unique_values.txt
src/lobster/assets/cortex_ab_tokenizer/__init__.py
src/lobster/assets/cortex_ab_tokenizer/special_tokens_map.json
src/lobster/assets/cortex_ab_tokenizer/tokenizer_config.json
src/lobster/assets/cortex_ab_tokenizer/vocab.txt
src/lobster/assets/hyena_tokenizer/__init__.py
src/lobster/assets/hyena_tokenizer/special_tokens_map.json
src/lobster/assets/hyena_tokenizer/tokenizer_config.json
src/lobster/assets/hyena_tokenizer/vocab.txt
src/lobster/assets/latent_generator_tokenizer/special_tokens_map.json
src/lobster/assets/latent_generator_tokenizer/tokenizer.json
src/lobster/assets/latent_generator_tokenizer/tokenizer_config.json
src/lobster/assets/latent_generator_tokenizer/vocab.txt
src/lobster/assets/mgm_tokenizer/__init__.py
src/lobster/assets/mgm_tokenizer/special_tokens_map.json
src/lobster/assets/mgm_tokenizer/tokenizer_config.json
src/lobster/assets/mgm_tokenizer/vocab.txt
src/lobster/assets/nucleotide_tokenizer/special_tokens_map.json
src/lobster/assets/nucleotide_tokenizer/tokenizer.json
src/lobster/assets/nucleotide_tokenizer/tokenizer_config.json
src/lobster/assets/pmlm_tokenizer/__init__.py
src/lobster/assets/pmlm_tokenizer/special_tokens_map.json
src/lobster/assets/pmlm_tokenizer/tokenizer_config.json
src/lobster/assets/pmlm_tokenizer/vocab.txt
src/lobster/assets/pmlm_tokenizer_32/__init__.py
src/lobster/assets/pmlm_tokenizer_32/special_tokens_map.json
src/lobster/assets/pmlm_tokenizer_32/tokenizer_config.json
src/lobster/assets/pmlm_tokenizer_32/vocab.txt
src/lobster/assets/smiles_tokenizer/special_tokens_map.json
src/lobster/assets/smiles_tokenizer/tokenizer.json
src/lobster/assets/smiles_tokenizer/tokenizer_config.json
src/lobster/assets/smiles_tokenizer/vocab.txt
src/lobster/assets/ume_tokenizers/amino_acid_tokenizer/special_tokens_map.json
src/lobster/assets/ume_tokenizers/amino_acid_tokenizer/tokenizer.json
src/lobster/assets/ume_tokenizers/amino_acid_tokenizer/tokenizer_config.json
src/lobster/assets/ume_tokenizers/amino_acid_tokenizer/vocab.txt
src/lobster/assets/ume_tokenizers/nucleotide_tokenizer/special_tokens_map.json
src/lobster/assets/ume_tokenizers/nucleotide_tokenizer/tokenizer.json
src/lobster/assets/ume_tokenizers/nucleotide_tokenizer/tokenizer_config.json
src/lobster/assets/ume_tokenizers/nucleotide_tokenizer/vocab.txt
src/lobster/assets/ume_tokenizers/smiles_tokenizer/special_tokens_map.json
src/lobster/assets/ume_tokenizers/smiles_tokenizer/tokenizer.json
src/lobster/assets/ume_tokenizers/smiles_tokenizer/tokenizer_config.json
src/lobster/assets/uniref_tokenzier/biological_process_unique_values_100.txt
src/lobster/assets/uniref_tokenzier/cellular_component_unique_values_100.txt
src/lobster/assets/uniref_tokenzier/cluster_name_unique_values_100.txt
src/lobster/assets/uniref_tokenzier/members_unique_values.txt
src/lobster/assets/uniref_tokenzier/molecular_function_unique_values_100.txt
src/lobster/assets/uniref_tokenzier/organism_unique_values_100.txt
src/lobster/assets/uniref_tokenzier/taxon_id_unique_values.txt
src/lobster/assets/uniref_tokenzier/taxon_unique_values_100.txt
src/lobster/callbacks/README.md
src/lobster/callbacks/__init__.py
src/lobster/callbacks/_calm_linear_probe_callback.py
src/lobster/callbacks/_dataloader_checkpoint_callback.py
src/lobster/callbacks/_dgeb_evaluation_callback.py
src/lobster/callbacks/_linear_probe_callback.py
src/lobster/callbacks/_moleculeace_linear_probe_callback.py
src/lobster/callbacks/_peer_evaluation_callback.py
src/lobster/callbacks/_peer_utils.py
src/lobster/callbacks/_perturbation_score_callback.py
src/lobster/callbacks/_tokens_per_second_callback.py
src/lobster/callbacks/_umap_visualization_callback.py
src/lobster/callbacks/_ume_grpo_logging_callback.py
src/lobster/cmdline/__init__.py
src/lobster/cmdline/_utils.py
src/lobster/cmdline/dgeb_eval.py
src/lobster/cmdline/embed.py
src/lobster/cmdline/eval_embed.py
src/lobster/cmdline/evaluate.py
src/lobster/cmdline/intervene.py
src/lobster/cmdline/intervene_multiproperty.py
src/lobster/cmdline/manage_ume_checkpoints.py
src/lobster/cmdline/perplexity.py
src/lobster/cmdline/predict.py
src/lobster/cmdline/train.py
src/lobster/concepts/__init__.py
src/lobster/concepts/_descriptors.py
src/lobster/concepts/_large_molecule_descriptors.py
src/lobster/concepts/_utils.py
src/lobster/constants/.gitkeep
src/lobster/constants/__init__.py
src/lobster/constants/_architecture_analyzer.py
src/lobster/constants/_calm_tasks.py
src/lobster/constants/_codon_table.py
src/lobster/constants/_hf.py
src/lobster/constants/_modality.py
src/lobster/constants/_moleculeace_tasks.py
src/lobster/constants/_peer_tasks.py
src/lobster/constants/_pooling.py
src/lobster/constants/_rdkit_descriptor_distributions.py
src/lobster/constants/_s3.py
src/lobster/constants/_scheduler_type.py
src/lobster/constants/_split.py
src/lobster/constants/_ume_models.py
src/lobster/constants/_weighted_concat_sampler_chunk_size.py
src/lobster/data/__init__.py
src/lobster/data/_calm_datamodule.py
src/lobster/data/_chembl_datamodule.py
src/lobster/data/_collate.py
src/lobster/data/_constants.py
src/lobster/data/_dataframe_dataset_in_memory.py
src/lobster/data/_dyab_data.py
src/lobster/data/_farthest_first_traversal.py
src/lobster/data/_fasta_datamodule.py
src/lobster/data/_m3_20m_datamodule.py
src/lobster/data/_minhasher.py
src/lobster/data/_mmseqs.py
src/lobster/data/_ppi_sequence_datamodule.py
src/lobster/data/_structure_datamodule.py
src/lobster/data/_ume_datamodule.py
src/lobster/data/_utils.py
src/lobster/datasets/__init__.py
src/lobster/datasets/_amplify_dataset.py
src/lobster/datasets/_atomica_dataset.py
src/lobster/datasets/_calm_dataset.py
src/lobster/datasets/_calm_property_dataset.py
src/lobster/datasets/_distributed_environment_utils.py
src/lobster/datasets/_fasta_dataset.py
src/lobster/datasets/_huggingface_iterable_dataset.py
src/lobster/datasets/_latent_generator_3d_coordinates_dataset.py
src/lobster/datasets/_m3_20m_dataset.py
src/lobster/datasets/_moleculeace_dataset.py
src/lobster/datasets/_multiplexed_sampling_dataset.py
src/lobster/datasets/_open_genome_2.py
src/lobster/datasets/_peer_dataset.py
src/lobster/datasets/_ptm_dataset.py
src/lobster/datasets/_round_robin_concat_iterable_dataset.py
src/lobster/datasets/_shuffled_iterable_dataset.py
src/lobster/datasets/_zinc_dataset.py
src/lobster/datasets/s3_datasets/README.md
src/lobster/datasets/s3_datasets/__init__.py
src/lobster/datasets/s3_datasets/_optimize.py
src/lobster/datasets/s3_datasets/amplify.py
src/lobster/datasets/s3_datasets/atomica.py
src/lobster/datasets/s3_datasets/base.py
src/lobster/datasets/s3_datasets/calm.py
src/lobster/datasets/s3_datasets/m320m.py
src/lobster/datasets/s3_datasets/peptide_atlas.py
src/lobster/datasets/s3_datasets/zinc.py
src/lobster/evaluation/README.md
src/lobster/evaluation/__init__.py
src/lobster/evaluation/_evaluate_model_with_callbacks.py
src/lobster/evaluation/_pooling_utils.py
src/lobster/evaluation/dgeb_adapter.py
src/lobster/evaluation/dgeb_mock_adapter.py
src/lobster/evaluation/dgeb_mock_runner.py
src/lobster/evaluation/dgeb_runner.py
src/lobster/evaluation/esm_dgeb_adapter.py
src/lobster/extern/openfold_utils/__init__.py
src/lobster/extern/openfold_utils/_data_pipeline.py
src/lobster/extern/openfold_utils/_data_transforms.py
src/lobster/extern/openfold_utils/_fape.py
src/lobster/extern/openfold_utils/_feats.py
src/lobster/extern/openfold_utils/_protein.py
src/lobster/extern/openfold_utils/_residue_constants.py
src/lobster/extern/openfold_utils/_rigids.py
src/lobster/extern/openfold_utils/_tensor_utils.py
src/lobster/extern/openfold_utils/resources/stereo_chemical_props.txt
src/lobster/features/__init__.py
src/lobster/features/_feature.py
src/lobster/hydra_config/__init__.py
src/lobster/hydra_config/embed.yaml
src/lobster/hydra_config/evaluate.yaml
src/lobster/hydra_config/intervene.yaml
src/lobster/hydra_config/intervene_multiproperty.yaml
src/lobster/hydra_config/manage_ume_checkpoints.yaml
src/lobster/hydra_config/perplexity.yaml
src/lobster/hydra_config/predict.yaml
src/lobster/hydra_config/train.yaml
src/lobster/hydra_config/callbacks/base.yaml
src/lobster/hydra_config/callbacks/batch_size_finder.yaml
src/lobster/hydra_config/callbacks/calm_linear_probe.yaml
src/lobster/hydra_config/callbacks/calm_linear_probe_fast.yaml
src/lobster/hydra_config/callbacks/default.yaml
src/lobster/hydra_config/callbacks/default_eval.yaml
src/lobster/hydra_config/callbacks/early_stopping.yaml
src/lobster/hydra_config/callbacks/lr_monitor.yaml
src/lobster/hydra_config/callbacks/model_checkpoint.yaml
src/lobster/hydra_config/callbacks/moleculeace_linear_probe.yaml
src/lobster/hydra_config/callbacks/moleculeace_linear_probe_fast.yaml
src/lobster/hydra_config/callbacks/naturalness.yaml
src/lobster/hydra_config/callbacks/perturbation_score.yaml
src/lobster/hydra_config/callbacks/progress_bar.yaml
src/lobster/hydra_config/callbacks/throughput.yaml
src/lobster/hydra_config/callbacks/timer.yaml
src/lobster/hydra_config/callbacks/tokens_per_second.yaml
src/lobster/hydra_config/callbacks/ume_grpo_logging.yaml
src/lobster/hydra_config/data/base.yaml
src/lobster/hydra_config/data/calm.yaml
src/lobster/hydra_config/data/cath.yaml
src/lobster/hydra_config/data/chembl.yaml
src/lobster/hydra_config/data/cmap.yaml
src/lobster/hydra_config/data/cyno_pk.yaml
src/lobster/hydra_config/data/dyab.yaml
src/lobster/hydra_config/data/fasta.yaml
src/lobster/hydra_config/data/m320m.yaml
src/lobster/hydra_config/data/ppi.yaml
src/lobster/hydra_config/data/ppi_infer.yaml
src/lobster/hydra_config/data/ume.yaml
src/lobster/hydra_config/data/transform_fn/hyena_tokenizer_transform.yaml
src/lobster/hydra_config/data/transform_fn/mgm_tokenizer_transform.yaml
src/lobster/hydra_config/data/transform_fn/nucleotide_tokenizer_transform.yaml
src/lobster/hydra_config/data/transform_fn/pmlm_tokenizer_transform.yaml
src/lobster/hydra_config/data/transform_fn/pt5_teacher_forcing_transform.yaml
src/lobster/hydra_config/data/transform_fn/smiles_tokenizer_transform.yaml
src/lobster/hydra_config/experiment/train_ume.yaml
src/lobster/hydra_config/experiment/ume-2/base.yaml
src/lobster/hydra_config/experiment/ume-2/small_molecule.yaml
src/lobster/hydra_config/logger/csv.yaml
src/lobster/hydra_config/logger/wandb.yaml
src/lobster/hydra_config/lr_scheduler/default.yaml
src/lobster/hydra_config/lr_scheduler/wsd.yaml
src/lobster/hydra_config/model/clip.yaml
src/lobster/hydra_config/model/clm.yaml
src/lobster/hydra_config/model/cmap.yaml
src/lobster/hydra_config/model/dyab.yaml
src/lobster/hydra_config/model/hyena.yaml
src/lobster/hydra_config/model/lobsterfold.yaml
src/lobster/hydra_config/model/mgm.yaml
src/lobster/hydra_config/model/mlm.yaml
src/lobster/hydra_config/model/mlp.yaml
src/lobster/hydra_config/model/modern_bert.yaml
src/lobster/hydra_config/model/neobert.yaml
src/lobster/hydra_config/model/ppi.yaml
src/lobster/hydra_config/model/seq2seq.yaml
src/lobster/hydra_config/model/ume.yaml
src/lobster/hydra_config/paths/default.yaml
src/lobster/hydra_config/plugins/base.yaml
src/lobster/hydra_config/plugins/bitsandbytes.yaml
src/lobster/hydra_config/plugins/default.yaml
src/lobster/hydra_config/setup/default.yaml
src/lobster/hydra_config/setup/seed/default.yaml
src/lobster/hydra_config/setup/torch/default.yaml
src/lobster/hydra_config/trainer/default.yaml
src/lobster/mcp/README.md
src/lobster/mcp/__init__.py
src/lobster/mcp/example_server.py
src/lobster/mcp/inference_server.py
src/lobster/mcp/manifest.json
src/lobster/mcp/package.json
src/lobster/mcp/requirements.txt
src/lobster/mcp/server.py
src/lobster/mcp/setup.py
src/lobster/mcp/tool_factory.py
src/lobster/mcp/models/__init__.py
src/lobster/mcp/models/config.py
src/lobster/mcp/tools/__init__.py
src/lobster/mcp/tools/concepts.py
src/lobster/mcp/tools/interventions.py
src/lobster/mcp/tools/representations.py
src/lobster/mcp/tools/tool_utils.py
src/lobster/metrics/__init__.py
src/lobster/metrics/_binary_classification.py
src/lobster/metrics/_perturbation_score.py
src/lobster/metrics/_random_neighbor_score.py
src/lobster/model/README.md
src/lobster/model/__init__.py
src/lobster/model/_cbmlm.py
src/lobster/model/_clip.py
src/lobster/model/_clm.py
src/lobster/model/_clm_configuration.py
src/lobster/model/_cmap.py
src/lobster/model/_conditioanalclassifiermlm.py
src/lobster/model/_conditioanalmlm.py
src/lobster/model/_disco_clip.py
src/lobster/model/_distributed_utils.py
src/lobster/model/_dyab.py
src/lobster/model/_linear_probe.py
src/lobster/model/_lobster_fold.py
src/lobster/model/_lobster_fold_base.py
src/lobster/model/_lobster_fold_configuration.py
src/lobster/model/_mgm.py
src/lobster/model/_mlm.py
src/lobster/model/_mlm_configuration.py
src/lobster/model/_mlp.py
src/lobster/model/_onnx_utils.py
src/lobster/model/_peft_lightning_module.py
src/lobster/model/_pooler.py
src/lobster/model/_pooling_layers.py
src/lobster/model/_ppi_clf.py
src/lobster/model/_seq2seq.py
src/lobster/model/_seq2seq_configuration.py
src/lobster/model/_ume.py
src/lobster/model/_utils.py
src/lobster/model/_utils_checkpoint.py
src/lobster/model/hyena/__init__.py
src/lobster/model/hyena/_hyena.py
src/lobster/model/hyena/_hyena_base.py
src/lobster/model/hyena/_hyena_configuration.py
src/lobster/model/integrations/__init__.py
src/lobster/model/integrations/ume_huggingface/README.md
src/lobster/model/integrations/ume_huggingface/__init__.py
src/lobster/model/integrations/ume_huggingface/configuration_ume.py
src/lobster/model/integrations/ume_huggingface/export_to_onnx.py
src/lobster/model/integrations/ume_huggingface/modeling_ume.py
src/lobster/model/integrations/ume_huggingface/register_model.py
src/lobster/model/integrations/ume_huggingface/tokenization_ume.py
src/lobster/model/integrations/ume_huggingface/upload_to_hub.py
src/lobster/model/integrations/ume_huggingface/upload_vocabs.py
src/lobster/model/integrations/ume_huggingface/model/README.md
src/lobster/model/latent_generator/README.md
src/lobster/model/latent_generator/__init__.py
src/lobster/model/latent_generator/reconstruction_results_table.md
src/lobster/model/latent_generator/callbacks/__init__.py
src/lobster/model/latent_generator/callbacks/_backbone_reconstruction.py
src/lobster/model/latent_generator/callbacks/_dssp_linear_probe.py
src/lobster/model/latent_generator/cmdline/__init__.py
src/lobster/model/latent_generator/cmdline/inference.py
src/lobster/model/latent_generator/cmdline/train.py
src/lobster/model/latent_generator/datamodules/__init__.py
src/lobster/model/latent_generator/datamodules/_structure_datamodule.py
src/lobster/model/latent_generator/datamodules/_utils.py
src/lobster/model/latent_generator/datasets/__init__.py
src/lobster/model/latent_generator/datasets/_ligand_dataset.py
src/lobster/model/latent_generator/datasets/_sampler.py
src/lobster/model/latent_generator/datasets/_structure_dataset.py
src/lobster/model/latent_generator/datasets/_structure_dataset_iterable.py
src/lobster/model/latent_generator/datasets/_transforms.py
src/lobster/model/latent_generator/example/example_pdbs/4erk_ligand.sdf
src/lobster/model/latent_generator/example/example_pdbs/4erk_protein.pdb
src/lobster/model/latent_generator/example/example_pdbs/7kdr_protein.pdb
src/lobster/model/latent_generator/hydra_config/__init__.py
src/lobster/model/latent_generator/hydra_config/train_multi.yaml
src/lobster/model/latent_generator/hydra_config/callbacks/backbone_reconstruction.yaml
src/lobster/model/latent_generator/hydra_config/callbacks/dssp_linear_probe.yaml
src/lobster/model/latent_generator/hydra_config/callbacks/lr_monitor.yaml
src/lobster/model/latent_generator/hydra_config/callbacks/model_checkpoint.yaml
src/lobster/model/latent_generator/hydra_config/callbacks/train_default.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_afdb.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_afdb_genie.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_ligand.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_pdb.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_pdb_pinder.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_pinder.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_pinder_3di.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/structure_pinder_esm.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/collate_fn/collate_fn_backbone.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/collate_fn/collate_fn_backbone_binder_target.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/collate_fn/collate_fn_ligand.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/ligand_transforms/structure_ligand_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/sampler/randomized_minority_upsampler.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/esm_embedding_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_3di_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_backbone_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_backbone_transform_with_template.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_backbone_with_esm_3di_c6d_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_backbone_with_esm_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_c6d_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_template_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_with_3di_c6d_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_with_3di_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_with_3di_zernike_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_with_c6d_transform.yaml
src/lobster/model/latent_generator/hydra_config/datamodule/transforms/structure_zernike_transform.yaml
src/lobster/model/latent_generator/hydra_config/hydra/default.yaml
src/lobster/model/latent_generator/hydra_config/logger/default.yaml
src/lobster/model/latent_generator/hydra_config/logger/wandb.yaml
src/lobster/model/latent_generator/hydra_config/paths/default.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/tokenizer_multi.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder_3di_c6d_sequence.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder_ligand.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder_ligand_3di_sequence.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder_ligand_aux.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder_ligand_element.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/struc_decoder_sequence.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/c6d_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/element_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/foldseek3di_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/rg_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/sasa_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/sequence_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/decoder_factory/decoders/vit_decoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/structure_losses.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/structure_losses_3di_c6d_sequence.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/structure_losses_ligand.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/structure_losses_ligand_aux.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/structure_losses_ligand_element.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/structure_losses_sequence.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/c6d_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/cce_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/element_cce_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/l2_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/ligand_l2_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/ligand_pairwise_l2_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/pairwise_l2_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/rg_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/sasa_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/loss_factory/losses/sequence_cce_loss.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/lr_scheduler/constant_lr.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/lr_scheduler/cosine_schedule_with_warmup.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/optim/adam.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/quantizer/slq_quantizer.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/quantizer/slq_quantizer_ligand.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/structure_encoder/vit_encoder.yaml
src/lobster/model/latent_generator/hydra_config/tokenizer/structure_encoder/vit_encoder_plm.yaml
src/lobster/model/latent_generator/io/__init__.py
src/lobster/model/latent_generator/io/_load_pdb.py
src/lobster/model/latent_generator/io/_write_pdb.py
src/lobster/model/latent_generator/models/__init__.py
src/lobster/model/latent_generator/models/vit/_vit_utils.py
src/lobster/model/latent_generator/quantizer/__init__.py
src/lobster/model/latent_generator/quantizer/_ligand_tokenizer.py
src/lobster/model/latent_generator/quantizer/_slq.py
src/lobster/model/latent_generator/structure_decoder/_3di_decoder.py
src/lobster/model/latent_generator/structure_decoder/__init__.py
src/lobster/model/latent_generator/structure_decoder/_c6d_decoder.py
src/lobster/model/latent_generator/structure_decoder/_decoder.py
src/lobster/model/latent_generator/structure_decoder/_decoder_factory.py
src/lobster/model/latent_generator/structure_decoder/_element_decoder.py
src/lobster/model/latent_generator/structure_decoder/_ligand_aux.py
src/lobster/model/latent_generator/structure_decoder/_seq_decoder.py
src/lobster/model/latent_generator/structure_decoder/_vit_decoder.py
src/lobster/model/latent_generator/structure_encoder/__init__.py
src/lobster/model/latent_generator/structure_encoder/_encoder.py
src/lobster/model/latent_generator/structure_encoder/_vit_encoder.py
src/lobster/model/latent_generator/structure_encoder/_vit_encoder_plm.py
src/lobster/model/latent_generator/tokenizer/__init__.py
src/lobster/model/latent_generator/tokenizer/_loss_factory.py
src/lobster/model/latent_generator/tokenizer/_losses.py
src/lobster/model/latent_generator/tokenizer/_tokenizer_multi.py
src/lobster/model/latent_generator/utils/__init__.py
src/lobster/model/latent_generator/utils/_gumbel.py
src/lobster/model/latent_generator/utils/_kinematics.py
src/lobster/model/latent_generator/utils/_lrf.py
src/lobster/model/latent_generator/utils/_molecular_frame.py
src/lobster/model/latent_generator/utils/_utils.py
src/lobster/model/latent_generator/utils/residue_constants.py
src/lobster/model/latent_generator/utils/mini3di/__init__.py
src/lobster/model/latent_generator/utils/mini3di/_encoder.py
src/lobster/model/latent_generator/utils/mini3di/_layers.py
src/lobster/model/latent_generator/utils/mini3di/_unkerasify.py
src/lobster/model/latent_generator/utils/mini3di/_utils.py
src/lobster/model/latent_generator/utils/mini3di/encoder_weights_3di.kerasify
src/lobster/model/llama_base/__init__.py
src/lobster/model/llama_base/_attn_mask_utils.py
src/lobster/model/llama_base/_llama_encoder.py
src/lobster/model/lm_base/__init__.py
src/lobster/model/lm_base/_lm_base.py
src/lobster/model/lm_base/_lm_base_heads.py
src/lobster/model/lm_base/_utils.py
src/lobster/model/losses/__init__.py
src/lobster/model/losses/_infonce_loss.py
src/lobster/model/losses/_symile_loss.py
src/lobster/model/modern_bert/LICENSE
src/lobster/model/modern_bert/__init__.py
src/lobster/model/modern_bert/_activation.py
src/lobster/model/modern_bert/_attention.py
src/lobster/model/modern_bert/_config.py
src/lobster/model/modern_bert/_embedding.py
src/lobster/model/modern_bert/_initialization.py
src/lobster/model/modern_bert/_layers.py
src/lobster/model/modern_bert/_mlp.py
src/lobster/model/modern_bert/_model.py
src/lobster/model/modern_bert/_modern_bert.py
src/lobster/model/modern_bert/_modern_bert_configuration.py
src/lobster/model/modern_bert/_normalization.py
src/lobster/model/modern_bert/_padding.py
src/lobster/model/modern_bert/_rotary.py
src/lobster/model/modern_bert/_utils.py
src/lobster/model/neobert/__init__.py
src/lobster/model/neobert/_config.py
src/lobster/model/neobert/_masking.py
src/lobster/model/neobert/_model.py
src/lobster/model/neobert/_rotary.py
src/lobster/model/neobert/_swiglu.py
src/lobster/model/neobert/neobert_lightning_module.py
src/lobster/model/neobert/neobert_module.py
src/lobster/model/openfold_utils/__init__.py
src/lobster/model/openfold_utils/_data_pipeline.py
src/lobster/model/openfold_utils/_data_transforms.py
src/lobster/model/openfold_utils/_fape.py
src/lobster/model/openfold_utils/_feats.py
src/lobster/model/openfold_utils/_protein.py
src/lobster/model/openfold_utils/_residue_constants.py
src/lobster/model/openfold_utils/_rigids.py
src/lobster/model/openfold_utils/_tensor_utils.py
src/lobster/model/openfold_utils/resources/stereo_chemical_props.txt
src/lobster/model/ume2/__init__.py
src/lobster/model/ume2/_ume_sequence_encoder.py
src/lobster/model/ume2/_ume_sequence_encoder_lightning_module.py
src/lobster/model/utils/__init__.py
src/lobster/model/utils/_architecture_analyzer.py
src/lobster/model/utils/_detect_modality.py
src/lobster/rl_training/README.md
src/lobster/rl_training/__init__.py
src/lobster/rl_training/reward_functions.py
src/lobster/rl_training/trainers.py
src/lobster/tokenization/__init__.py
src/lobster/tokenization/_amino_acid.py
src/lobster/tokenization/_cached_bert_tokenizer.py
src/lobster/tokenization/_hyena_tokenizer.py
src/lobster/tokenization/_hyena_tokenizer_transform.py
src/lobster/tokenization/_latent_generator_3d_coord_tokenizer.py
src/lobster/tokenization/_load_vocab_file.py
src/lobster/tokenization/_make_pretrained_tokenizer_fast.py
src/lobster/tokenization/_mgm_tokenizer.py
src/lobster/tokenization/_mgm_tokenizer_transform.py
src/lobster/tokenization/_nucleotide_tokenizer.py
src/lobster/tokenization/_pmlm_custom_concept_tokenizer_transform.py
src/lobster/tokenization/_pmlm_tokenizer.py
src/lobster/tokenization/_pmlm_tokenizer_transform.py
src/lobster/tokenization/_smiles_tokenizer.py
src/lobster/tokenization/_ume_tokenizers.py
src/lobster/transforms/__init__.py
src/lobster/transforms/_atom3d_ppi_transforms.py
src/lobster/transforms/_auto_tokenizer_transform.py
src/lobster/transforms/_binarize.py
src/lobster/transforms/_equivalence_transforms.py
src/lobster/transforms/_lambda.py
src/lobster/transforms/_modality_aware_transform.py
src/lobster/transforms/_rdkit_descriptors.py
src/lobster/transforms/_structure_featurizer.py
src/lobster/transforms/_tokenizer_transform.py
src/lobster/transforms/_transform.py
src/lobster/transforms/functional/__init__.py
src/lobster/transforms/functional/_biopython.py
src/lobster/transforms/functional/_convert_seqs.py
src/lobster/transforms/functional/_rdkit_descs.py
src/lobster/transforms/functional/_sample_item.py
src/lobster/transforms/functional/_sample_tokenized_input.py
src/lobster/transforms/functional/_utils.py
static/css/bulma-carousel.min.css
static/css/bulma-slider.min.css
static/css/bulma.css.map.txt
static/css/bulma.min.css
static/css/fontawesome.all.min.css
static/css/index.css
static/images/carousel1.png
static/images/carousel2.png
static/images/favicon.ico
static/js/bulma-carousel.js
static/js/bulma-carousel.min.js
static/js/bulma-slider.js
static/js/bulma-slider.min.js
static/js/fontawesome.all.min.js
static/js/index.js
static/pdfs/Cramming_MLSB_Neurips_2023_poster.pdf
static/videos/banner_video.mp4
static/videos/carousel1.mp4
static/videos/carousel2.mp4
static/videos/carousel3.mp4
test_data/2ah5A02.pdb
test_data/fv.pdb
test_data/query.fasta
test_data/test.pdb
test_data/parquet/data.parquet
test_data/pdbs/1a2oA01
test_data/pdbs/1ae2A00
tests/__init__.py
tests/test_placeholder.py
tests/lobster/__init__.py
tests/lobster/conftest.py
tests/lobster/test__imports.py
tests/lobster/callbacks/test__calm_linear_probe_callback.py
tests/lobster/callbacks/test__dataloader_checkpoint_callback.py
tests/lobster/callbacks/test__dgeb_evaluation_callback.py
tests/lobster/callbacks/test__peer_evaluation_callback.py
tests/lobster/callbacks/test__perturbation_score_callback.py
tests/lobster/callbacks/test__tokens_per_second_callback.py
tests/lobster/callbacks/test__umap_visualization_callback.py
tests/lobster/cmdline/__init__.py
tests/lobster/cmdline/test__cmdline.py
tests/lobster/data/__init__.py
tests/lobster/data/test__calm_datamodule.py
tests/lobster/data/test__chembl_datamodule.py
tests/lobster/data/test__farthest_first_traversal.py
tests/lobster/data/test__fasta_lightning_datamodule.py
tests/lobster/data/test__minhasher.py
tests/lobster/data/test__mmseqs.py
tests/lobster/data/test__ume_datamodule.py
tests/lobster/datasets/__init__.py
tests/lobster/datasets/test__amplify_dataset.py
tests/lobster/datasets/test__atomica_dataset.py
tests/lobster/datasets/test__calm_dataset.py
tests/lobster/datasets/test__concat_iterable_dataset.py
tests/lobster/datasets/test__huggingface_iterable_dataset.py
tests/lobster/datasets/test__latent_generator_3d_coord_dataset.py
tests/lobster/datasets/test__m3_20m_dataset.py
tests/lobster/datasets/test__multiplexed_sampling_dataset.py
tests/lobster/datasets/test__ptm_dataset.py
tests/lobster/datasets/test__shuffled_iterable_dataset.py
tests/lobster/datasets/test__zinc_dataset.py
tests/lobster/evaluation/test_dgeb_adapter.py
tests/lobster/evaluation/test_dgeb_integration.py
tests/lobster/evaluation/test_esm_dgeb_adapter.py
tests/lobster/evaluation/test_evaluate_model_with_callbacks.py
tests/lobster/evaluation/test_mock_dgeb.py
tests/lobster/mcp/__init__.py
tests/lobster/mcp/test_concepts.py
tests/lobster/mcp/test_interventions.py
tests/lobster/mcp/test_representations.py
tests/lobster/mcp/test_server.py
tests/lobster/mcp/test_simple_server.py
tests/lobster/mcp/test_tool_factory.py
tests/lobster/mcp/test_tool_utils.py
tests/lobster/metrics/test__perturbation_score_metrics.py
tests/lobster/metrics/test__random_neighbor_score.py
tests/lobster/model/__init__.py
tests/lobster/model/test__cbm.py
tests/lobster/model/test__clm.py
tests/lobster/model/test__dyab.py
tests/lobster/model/test__foldseek_transform.py
tests/lobster/model/test__linear_probe.py
tests/lobster/model/test__lobsterfold.py
tests/lobster/model/test__mgm.py
tests/lobster/model/test__mlm.py
tests/lobster/model/test__mlp.py
tests/lobster/model/test__peft_lightning_module.py
tests/lobster/model/test__pooler.py
tests/lobster/model/test__ppi.py
tests/lobster/model/test__ume.py
tests/lobster/model/test__ume_onnx.py
tests/lobster/model/test__utils_checkpoint.py
tests/lobster/model/hyena/__init__.py
tests/lobster/model/hyena/test__hyena.py
tests/lobster/model/latent_generator/test_latent_generator_recon.py
tests/lobster/model/modern_bert/test__modern_bert.py
tests/lobster/model/modern_bert/test_rotary.py
tests/lobster/model/neobert/test__masking.py
tests/lobster/model/neobert/test_neobert_lightning_module.py
tests/lobster/model/neobert/test_neobert_module.py
tests/lobster/model/openfold_utils/test__fape.py
tests/lobster/model/ume2/test_sequence_encoder_lightning_module.py
tests/lobster/model/utils/test__architecture_analyzer.py
tests/lobster/model/utils/test__detect_modality.py
tests/lobster/model/utils/test_detect_modality.py
tests/lobster/rl_training/__init__.py
tests/lobster/rl_training/test_reward_functions.py
tests/lobster/rl_training/test_trainers.py
tests/lobster/tokenization/__init__.py
tests/lobster/tokenization/test__hyena_tokenizer.py
tests/lobster/tokenization/test__hyena_tokenizer_transform.py
tests/lobster/tokenization/test__latent_generator_tokenizer.py
tests/lobster/tokenization/test__make_pretrained_tokenizer_fast.py
tests/lobster/tokenization/test__mgm_tokenizer.py
tests/lobster/tokenization/test__mgm_tokenizer_transform.py
tests/lobster/tokenization/test__nucleotide_tokenizer.py
tests/lobster/tokenization/test__pmlm_tokenizer.py
tests/lobster/tokenization/test__smiles_tokenizer.py
tests/lobster/tokenization/test__ume_tokenizers.py
tests/lobster/transforms/__init__.py
tests/lobster/transforms/test__binarize_transform.py
tests/lobster/transforms/test__equivalence_transforms.py
tests/lobster/transforms/test__structure_featurizer.py
tests/lobster/transforms/test__tokenizer_transform.py
tests/lobster/transforms/test_modality_aware_transform.py
tests/lobster/transforms/functional/test__convert_seqs.py
tests/lobster/transforms/functional/test__rdkit_descs.py
tests/lobster/transforms/functional/test__utils.py