LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
src/gt4sd/__init__.py
src/gt4sd/configuration.py
src/gt4sd/conftest.py
src/gt4sd/exceptions.py
src/gt4sd/py.typed
src/gt4sd/s3.py
src/gt4sd.egg-info/PKG-INFO
src/gt4sd.egg-info/SOURCES.txt
src/gt4sd.egg-info/dependency_links.txt
src/gt4sd.egg-info/entry_points.txt
src/gt4sd.egg-info/requires.txt
src/gt4sd.egg-info/top_level.txt
src/gt4sd/algorithms/__init__.py
src/gt4sd/algorithms/core.py
src/gt4sd/algorithms/registry.py
src/gt4sd/algorithms/conditional_generation/__init__.py
src/gt4sd/algorithms/conditional_generation/guacamol/__init__.py
src/gt4sd/algorithms/conditional_generation/guacamol/core.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/__init__.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/graph_ga.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/graph_mcts.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/moses_aae.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/moses_organ.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/moses_vae.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/smiles_ga.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/smiles_lstm_hc.py
src/gt4sd/algorithms/conditional_generation/guacamol/implementation/smiles_lstm_ppo.py
src/gt4sd/algorithms/conditional_generation/key_bert/__init__.py
src/gt4sd/algorithms/conditional_generation/key_bert/core.py
src/gt4sd/algorithms/conditional_generation/key_bert/implementation.py
src/gt4sd/algorithms/conditional_generation/molgx/__init__.py
src/gt4sd/algorithms/conditional_generation/molgx/core.py
src/gt4sd/algorithms/conditional_generation/molgx/implementation.py
src/gt4sd/algorithms/conditional_generation/paccmann_rl/__init__.py
src/gt4sd/algorithms/conditional_generation/paccmann_rl/core.py
src/gt4sd/algorithms/conditional_generation/paccmann_rl/implementation.py
src/gt4sd/algorithms/conditional_generation/regression_transformer/__init__.py
src/gt4sd/algorithms/conditional_generation/regression_transformer/core.py
src/gt4sd/algorithms/conditional_generation/regression_transformer/implementation.py
src/gt4sd/algorithms/conditional_generation/regression_transformer/utils.py
src/gt4sd/algorithms/conditional_generation/reinvent/__init__.py
src/gt4sd/algorithms/conditional_generation/reinvent/core.py
src/gt4sd/algorithms/conditional_generation/reinvent/implementation.py
src/gt4sd/algorithms/conditional_generation/reinvent/reinvent_core/__init__.py
src/gt4sd/algorithms/conditional_generation/reinvent/reinvent_core/core.py
src/gt4sd/algorithms/conditional_generation/template/__init__.py
src/gt4sd/algorithms/conditional_generation/template/core.py
src/gt4sd/algorithms/conditional_generation/template/implementation.py
src/gt4sd/algorithms/conditional_generation/tests/__init__.py
src/gt4sd/algorithms/conditional_generation/tests/test_guacamol.py
src/gt4sd/algorithms/conditional_generation/tests/test_key_bert.py
src/gt4sd/algorithms/conditional_generation/tests/test_molgx.py
src/gt4sd/algorithms/conditional_generation/tests/test_moses.py
src/gt4sd/algorithms/conditional_generation/tests/test_paccmann_rl.py
src/gt4sd/algorithms/conditional_generation/tests/test_regression_transformer.py
src/gt4sd/algorithms/conditional_generation/tests/test_reinvent.py
src/gt4sd/algorithms/controlled_sampling/__init__.py
src/gt4sd/algorithms/controlled_sampling/advanced_manufacturing/__init__.py
src/gt4sd/algorithms/controlled_sampling/advanced_manufacturing/core.py
src/gt4sd/algorithms/controlled_sampling/advanced_manufacturing/implementation/__init__.py
src/gt4sd/algorithms/controlled_sampling/advanced_manufacturing/implementation/core.py
src/gt4sd/algorithms/controlled_sampling/advanced_manufacturing/implementation/nccr/__init__.py
src/gt4sd/algorithms/controlled_sampling/advanced_manufacturing/implementation/nccr/core.py
src/gt4sd/algorithms/controlled_sampling/class_controlled_sampling/__init__.py
src/gt4sd/algorithms/controlled_sampling/class_controlled_sampling/core.py
src/gt4sd/algorithms/controlled_sampling/paccmann_gp/__init__.py
src/gt4sd/algorithms/controlled_sampling/paccmann_gp/core.py
src/gt4sd/algorithms/controlled_sampling/paccmann_gp/implementation.py
src/gt4sd/algorithms/controlled_sampling/tests/__init__.py
src/gt4sd/algorithms/controlled_sampling/tests/test_advanced_manufacturing.py
src/gt4sd/algorithms/controlled_sampling/tests/test_class_controlled_sampling.py
src/gt4sd/algorithms/controlled_sampling/tests/test_paccmann_gp.py
src/gt4sd/algorithms/generation/__init__.py
src/gt4sd/algorithms/generation/diffusion/__init__.py
src/gt4sd/algorithms/generation/diffusion/core.py
src/gt4sd/algorithms/generation/diffusion/implementation.py
src/gt4sd/algorithms/generation/diffusion/geodiff/__init__.py
src/gt4sd/algorithms/generation/diffusion/geodiff/core.py
src/gt4sd/algorithms/generation/diffusion/geodiff/model/__init__.py
src/gt4sd/algorithms/generation/diffusion/geodiff/model/core.py
src/gt4sd/algorithms/generation/diffusion/geodiff/model/layers.py
src/gt4sd/algorithms/generation/diffusion/geodiff/model/utils.py
src/gt4sd/algorithms/generation/hugging_face/__init__.py
src/gt4sd/algorithms/generation/hugging_face/core.py
src/gt4sd/algorithms/generation/hugging_face/implementation.py
src/gt4sd/algorithms/generation/moler/__init__.py
src/gt4sd/algorithms/generation/moler/core.py
src/gt4sd/algorithms/generation/moler/implementation.py
src/gt4sd/algorithms/generation/paccmann_vae/__init__.py
src/gt4sd/algorithms/generation/paccmann_vae/core.py
src/gt4sd/algorithms/generation/paccmann_vae/implementation.py
src/gt4sd/algorithms/generation/pgt/__init__.py
src/gt4sd/algorithms/generation/pgt/core.py
src/gt4sd/algorithms/generation/pgt/implementation.py
src/gt4sd/algorithms/generation/polymer_blocks/__init__.py
src/gt4sd/algorithms/generation/polymer_blocks/core.py
src/gt4sd/algorithms/generation/polymer_blocks/implementation.py
src/gt4sd/algorithms/generation/tests/__init__.py
src/gt4sd/algorithms/generation/tests/mol_dct.pkl
src/gt4sd/algorithms/generation/tests/test_diffusion.py
src/gt4sd/algorithms/generation/tests/test_hugging_face.py
src/gt4sd/algorithms/generation/tests/test_moler.py
src/gt4sd/algorithms/generation/tests/test_paccmann_vae.py
src/gt4sd/algorithms/generation/tests/test_pgt.py
src/gt4sd/algorithms/generation/tests/test_polymer_blocks.py
src/gt4sd/algorithms/generation/tests/test_torchdrug.py
src/gt4sd/algorithms/generation/torchdrug/__init__.py
src/gt4sd/algorithms/generation/torchdrug/core.py
src/gt4sd/algorithms/generation/torchdrug/implementation.py
src/gt4sd/algorithms/prediction/__init__.py
src/gt4sd/algorithms/prediction/paccmann/__init__.py
src/gt4sd/algorithms/prediction/paccmann/core.py
src/gt4sd/algorithms/prediction/paccmann/implementation.py
src/gt4sd/algorithms/prediction/tests/__init__.py
src/gt4sd/algorithms/prediction/tests/test_paccmann.py
src/gt4sd/algorithms/prediction/tests/test_topics_zero_shot.py
src/gt4sd/algorithms/prediction/topics_zero_shot/__init__.py
src/gt4sd/algorithms/prediction/topics_zero_shot/core.py
src/gt4sd/algorithms/prediction/topics_zero_shot/implementation.py
src/gt4sd/algorithms/tests/__init__.py
src/gt4sd/algorithms/tests/test_config.py
src/gt4sd/algorithms/tests/test_registry.py
src/gt4sd/cli/__init__.py
src/gt4sd/cli/algorithms.py
src/gt4sd/cli/argument_parser.py
src/gt4sd/cli/hf_to_st_converter.py
src/gt4sd/cli/inference.py
src/gt4sd/cli/load_arguments_from_dataclass.py
src/gt4sd/cli/saving.py
src/gt4sd/cli/trainer.py
src/gt4sd/cli/upload.py
src/gt4sd/domains/__init__.py
src/gt4sd/domains/core.py
src/gt4sd/domains/materials/__init__.py
src/gt4sd/domains/materials/protein_encoding.py
src/gt4sd/domains/tests/__init__.py
src/gt4sd/extras/__init__.py
src/gt4sd/frameworks/__init__.py
src/gt4sd/frameworks/cgcnn/__init__.py
src/gt4sd/frameworks/cgcnn/data.py
src/gt4sd/frameworks/cgcnn/model.py
src/gt4sd/frameworks/crystals_rfc/__init__.py
src/gt4sd/frameworks/crystals_rfc/atomic_data.csv
src/gt4sd/frameworks/crystals_rfc/feature_engine.py
src/gt4sd/frameworks/crystals_rfc/rf_classifier.py
src/gt4sd/frameworks/enzeptional/__init__.py
src/gt4sd/frameworks/enzeptional/tests/__init__.py
src/gt4sd/frameworks/enzeptional/tests/test_core.py
src/gt4sd/frameworks/enzeptional/tests/test_processing.py
src/gt4sd/frameworks/gflownet/__init__.py
src/gt4sd/frameworks/gflownet/util.py
src/gt4sd/frameworks/gflownet/arg_parser/__init__.py
src/gt4sd/frameworks/gflownet/arg_parser/parser.py
src/gt4sd/frameworks/gflownet/arg_parser/utils.py
src/gt4sd/frameworks/gflownet/dataloader/__init__.py
src/gt4sd/frameworks/gflownet/dataloader/data_module.py
src/gt4sd/frameworks/gflownet/dataloader/dataset.py
src/gt4sd/frameworks/gflownet/dataloader/sampler.py
src/gt4sd/frameworks/gflownet/envs/__init__.py
src/gt4sd/frameworks/gflownet/envs/graph_building_env.py
src/gt4sd/frameworks/gflownet/envs/mol_building_env.py
src/gt4sd/frameworks/gflownet/loss/__init__.py
src/gt4sd/frameworks/gflownet/loss/td_loss.py
src/gt4sd/frameworks/gflownet/loss/trajectory_balance.py
src/gt4sd/frameworks/gflownet/ml/__init__.py
src/gt4sd/frameworks/gflownet/ml/module.py
src/gt4sd/frameworks/gflownet/ml/models/__init__.py
src/gt4sd/frameworks/gflownet/ml/models/graph_transformer.py
src/gt4sd/frameworks/gflownet/ml/models/mxmnet.py
src/gt4sd/frameworks/gflownet/tests/__init__.py
src/gt4sd/frameworks/gflownet/tests/qm9.py
src/gt4sd/frameworks/gflownet/tests/test_gfn.py
src/gt4sd/frameworks/gflownet/train/__init__.py
src/gt4sd/frameworks/gflownet/train/core.py
src/gt4sd/frameworks/granular/__init__.py
src/gt4sd/frameworks/granular/arg_parser/__init__.py
src/gt4sd/frameworks/granular/arg_parser/parser.py
src/gt4sd/frameworks/granular/arg_parser/utils.py
src/gt4sd/frameworks/granular/dataloader/__init__.py
src/gt4sd/frameworks/granular/dataloader/data_module.py
src/gt4sd/frameworks/granular/dataloader/dataset.py
src/gt4sd/frameworks/granular/dataloader/sampler.py
src/gt4sd/frameworks/granular/ml/__init__.py
src/gt4sd/frameworks/granular/ml/module.py
src/gt4sd/frameworks/granular/ml/models/__init__.py
src/gt4sd/frameworks/granular/ml/models/activation.py
src/gt4sd/frameworks/granular/ml/models/base_model.py
src/gt4sd/frameworks/granular/ml/models/loss.py
src/gt4sd/frameworks/granular/ml/models/model_builder.py
src/gt4sd/frameworks/granular/ml/models/module.py
src/gt4sd/frameworks/granular/ml/models/utils.py
src/gt4sd/frameworks/granular/ml/models/mlp_auto_encoder/__init__.py
src/gt4sd/frameworks/granular/ml/models/mlp_auto_encoder/core.py
src/gt4sd/frameworks/granular/ml/models/mlp_predictor/__init__.py
src/gt4sd/frameworks/granular/ml/models/mlp_predictor/core.py
src/gt4sd/frameworks/granular/ml/models/no_encoding/__init__.py
src/gt4sd/frameworks/granular/ml/models/no_encoding/core.py
src/gt4sd/frameworks/granular/ml/models/vae_mlp/__init__.py
src/gt4sd/frameworks/granular/ml/models/vae_mlp/core.py
src/gt4sd/frameworks/granular/ml/models/vae_rnn/__init__.py
src/gt4sd/frameworks/granular/ml/models/vae_rnn/core.py
src/gt4sd/frameworks/granular/ml/models/vae_trans/__init__.py
src/gt4sd/frameworks/granular/ml/models/vae_trans/core.py
src/gt4sd/frameworks/granular/tests/__init__.py
src/gt4sd/frameworks/granular/tests/test_tokenizer.py
src/gt4sd/frameworks/granular/tokenizer/__init__.py
src/gt4sd/frameworks/granular/tokenizer/tokenizer.py
src/gt4sd/frameworks/granular/train/__init__.py
src/gt4sd/frameworks/granular/train/core.py
src/gt4sd/frameworks/torch/__init__.py
src/gt4sd/frameworks/torch/vae.py
src/gt4sd/properties/__init__.py
src/gt4sd/properties/core.py
src/gt4sd/properties/scorer.py
src/gt4sd/properties/utils.py
src/gt4sd/properties/crystals/__init__.py
src/gt4sd/properties/crystals/core.py
src/gt4sd/properties/molecules/__init__.py
src/gt4sd/properties/molecules/core.py
src/gt4sd/properties/molecules/functions.py
src/gt4sd/properties/proteins/__init__.py
src/gt4sd/properties/proteins/core.py
src/gt4sd/properties/proteins/functions.py
src/gt4sd/properties/scores/__init__.py
src/gt4sd/properties/scores/core.py
src/gt4sd/properties/tests/__init__.py
src/gt4sd/properties/tests/test_properties.py
src/gt4sd/properties/tests/test_properties_scorer.py
src/gt4sd/tests/__init__.py
src/gt4sd/tests/test_configuration.py
src/gt4sd/tests/test_exceptions.py
src/gt4sd/tests/test_s3.py
src/gt4sd/tests/utils.py
src/gt4sd/training_pipelines/__init__.py
src/gt4sd/training_pipelines/core.py
src/gt4sd/training_pipelines/mock_training_pipeline.json
src/gt4sd/training_pipelines/terminator_training.json
src/gt4sd/training_pipelines/cgcnn/__init__.py
src/gt4sd/training_pipelines/cgcnn/core.py
src/gt4sd/training_pipelines/crystals_crf/__init__.py
src/gt4sd/training_pipelines/crystals_crf/core.py
src/gt4sd/training_pipelines/diffusion/__init__.py
src/gt4sd/training_pipelines/diffusion/core.py
src/gt4sd/training_pipelines/guacamol_baselines/__init__.py
src/gt4sd/training_pipelines/guacamol_baselines/core.py
src/gt4sd/training_pipelines/guacamol_baselines/smiles_lstm/__init__.py
src/gt4sd/training_pipelines/guacamol_baselines/smiles_lstm/core.py
src/gt4sd/training_pipelines/moses/__init__.py
src/gt4sd/training_pipelines/moses/core.py
src/gt4sd/training_pipelines/moses/organ/__init__.py
src/gt4sd/training_pipelines/moses/organ/core.py
src/gt4sd/training_pipelines/moses/vae/__init__.py
src/gt4sd/training_pipelines/moses/vae/core.py
src/gt4sd/training_pipelines/paccmann/__init__.py
src/gt4sd/training_pipelines/paccmann/core.py
src/gt4sd/training_pipelines/paccmann/vae/__init__.py
src/gt4sd/training_pipelines/paccmann/vae/core.py
src/gt4sd/training_pipelines/pytorch_lightning/__init__.py
src/gt4sd/training_pipelines/pytorch_lightning/gflownet/__init__.py
src/gt4sd/training_pipelines/pytorch_lightning/gflownet/core.py
src/gt4sd/training_pipelines/pytorch_lightning/granular/__init__.py
src/gt4sd/training_pipelines/pytorch_lightning/granular/core.py
src/gt4sd/training_pipelines/pytorch_lightning/molformer/__init__.py
src/gt4sd/training_pipelines/pytorch_lightning/molformer/core.py
src/gt4sd/training_pipelines/regression_transformer/__init__.py
src/gt4sd/training_pipelines/regression_transformer/core.py
src/gt4sd/training_pipelines/regression_transformer/implementation.py
src/gt4sd/training_pipelines/regression_transformer/utils.py
src/gt4sd/training_pipelines/tests/__init__.py
src/gt4sd/training_pipelines/tests/molecules.smi
src/gt4sd/training_pipelines/tests/test_argument_parser.py
src/gt4sd/training_pipelines/tests/test_training_cgnn.py
src/gt4sd/training_pipelines/tests/test_training_crystals_rfc.py
src/gt4sd/training_pipelines/tests/test_training_diffusion.py
src/gt4sd/training_pipelines/tests/test_training_gflownet.py
src/gt4sd/training_pipelines/tests/test_training_guacamol_baselines_smiles_lstm.py
src/gt4sd/training_pipelines/tests/test_training_moses_organ.py
src/gt4sd/training_pipelines/tests/test_training_moses_vae.py
src/gt4sd/training_pipelines/tests/test_training_paccmann_vae.py
src/gt4sd/training_pipelines/tests/test_training_pipelines.py
src/gt4sd/training_pipelines/tests/test_training_regression_transformer.py
src/gt4sd/training_pipelines/tests/test_training_torchdrug_gcpn.py
src/gt4sd/training_pipelines/tests/test_training_torchdrug_graphaf.py
src/gt4sd/training_pipelines/torchdrug/__init__.py
src/gt4sd/training_pipelines/torchdrug/core.py
src/gt4sd/training_pipelines/torchdrug/dataset.py
src/gt4sd/training_pipelines/torchdrug/unpatch.py
src/gt4sd/training_pipelines/torchdrug/gcpn/__init__.py
src/gt4sd/training_pipelines/torchdrug/gcpn/core.py
src/gt4sd/training_pipelines/torchdrug/graphaf/__init__.py
src/gt4sd/training_pipelines/torchdrug/graphaf/core.py