.gitignore
1agent_sweep.sh
LICENSE
README.md
cleanup_files.sh
codecov.yml
enable_ipu.sh
env.yml
install_ipu.sh
mkdocs.yml
pyproject.toml
requirements_ipu.txt
.github/CODEOWNERS
.github/CODE_OF_CONDUCT.md
.github/PULL_REQUEST_TEMPLATE.md
.github/workflows/code-check.yml
.github/workflows/doc.yml
.github/workflows/release.yml
.github/workflows/test.yml
.github/workflows/test_ipu.yml
docs/baseline.md
docs/contribute.md
docs/dataset_abstract.png
docs/datasets.md
docs/design.md
docs/index.md
docs/license.md
docs/pretrained_models.md
docs/_assets/css/custom-graphium.css
docs/_assets/css/custom.css
docs/_assets/js/google-analytics.js
docs/api/graphium.config.md
docs/api/graphium.data.md
docs/api/graphium.features.md
docs/api/graphium.finetuning.md
docs/api/graphium.ipu.md
docs/api/graphium.trainer.md
docs/api/graphium.utils.md
docs/api/graphium.visualization.md
docs/api/graphium.nn/architectures.md
docs/api/graphium.nn/encoders.md
docs/api/graphium.nn/graphium.nn.md
docs/api/graphium.nn/pyg_layers.md
docs/cli/graphium-train.md
docs/cli/graphium.md
docs/cli/reference.md
docs/images/datamodule.png
docs/images/full_graph_network.png
docs/images/logo.png
docs/images/logo.svg
docs/tutorials/feature_processing/add_new_positional_encoding.ipynb
docs/tutorials/feature_processing/choosing_parallelization.ipynb
docs/tutorials/feature_processing/csv_to_parquet.ipynb
docs/tutorials/feature_processing/timing_parallel.ipynb
docs/tutorials/gnn/add_new_gnn_layers.ipynb
docs/tutorials/gnn/making_gnn_networks.ipynb
docs/tutorials/gnn/using_gnn_layers.ipynb
docs/tutorials/model_training/running-multitask-ipu.ipynb
docs/tutorials/model_training/simple-molecular-model.ipynb
expts/__init__.py
expts/dataset_benchmark.py
expts/debug_yaml.py
expts/main_run_get_fingerprints.py
expts/main_run_multitask.py
expts/main_run_predict.py
expts/main_run_test.py
expts/run_validation_test.py
expts/configs/config_gps_10M_pcqm4m.yaml
expts/configs/config_gps_10M_pcqm4m_mod.yaml
expts/configs/config_mpnn_10M_b3lyp.yaml
expts/configs/config_mpnn_pcqm4m.yaml
expts/data/micro_zinc_splits.csv
expts/data/tiny_zinc_splits.csv
expts/hydra-configs/README.md
expts/hydra-configs/__init__.py
expts/hydra-configs/main.yaml
expts/hydra-configs/accelerator/cpu.yaml
expts/hydra-configs/accelerator/gpu.yaml
expts/hydra-configs/accelerator/ipu.yaml
expts/hydra-configs/architecture/largemix.yaml
expts/hydra-configs/architecture/pcqm4m.yaml
expts/hydra-configs/architecture/toymix.yaml
expts/hydra-configs/experiment/toymix_mpnn.yaml
expts/hydra-configs/finetuning/admet.yaml
expts/hydra-configs/hparam_search/optuna.yaml
expts/hydra-configs/model/gcn.yaml
expts/hydra-configs/model/gin.yaml
expts/hydra-configs/model/gine.yaml
expts/hydra-configs/model/gpspp.yaml
expts/hydra-configs/model/mpnn.yaml
expts/hydra-configs/tasks/admet.yaml
expts/hydra-configs/tasks/l1000_mcf7.yaml
expts/hydra-configs/tasks/l1000_vcap.yaml
expts/hydra-configs/tasks/largemix.yaml
expts/hydra-configs/tasks/pcba_1328.yaml
expts/hydra-configs/tasks/pcqm4m.yaml
expts/hydra-configs/tasks/pcqm4m_g25.yaml
expts/hydra-configs/tasks/pcqm4m_n4.yaml
expts/hydra-configs/tasks/toymix.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/admet.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/l1000_mcf7.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/l1000_vcap.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/largemix.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/pcba_1328.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/pcqm4m.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/pcqm4m_g25.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/pcqm4m_n4.yaml
expts/hydra-configs/tasks/loss_metrics_datamodule/toymix.yaml
expts/hydra-configs/tasks/task_heads/admet.yaml
expts/hydra-configs/tasks/task_heads/l1000_mcf7.yaml
expts/hydra-configs/tasks/task_heads/l1000_vcap.yaml
expts/hydra-configs/tasks/task_heads/largemix.yaml
expts/hydra-configs/tasks/task_heads/pcba_1328.yaml
expts/hydra-configs/tasks/task_heads/pcqm4m.yaml
expts/hydra-configs/tasks/task_heads/pcqm4m_g25.yaml
expts/hydra-configs/tasks/task_heads/pcqm4m_n4.yaml
expts/hydra-configs/tasks/task_heads/toymix.yaml
expts/hydra-configs/training/largemix.yaml
expts/hydra-configs/training/pcqm4m.yaml
expts/hydra-configs/training/toymix.yaml
expts/hydra-configs/training/accelerator/largemix_cpu.yaml
expts/hydra-configs/training/accelerator/largemix_gpu.yaml
expts/hydra-configs/training/accelerator/largemix_ipu.yaml
expts/hydra-configs/training/accelerator/pcqm4m_ipu.yaml
expts/hydra-configs/training/accelerator/toymix_cpu.yaml
expts/hydra-configs/training/accelerator/toymix_gpu.yaml
expts/hydra-configs/training/accelerator/toymix_ipu.yaml
expts/hydra-configs/training/model/largemix_gcn.yaml
expts/hydra-configs/training/model/largemix_gin.yaml
expts/hydra-configs/training/model/largemix_gine.yaml
expts/hydra-configs/training/model/pcqm4m_gpspp.yaml
expts/hydra-configs/training/model/pcqm4m_mpnn.yaml
expts/hydra-configs/training/model/toymix_gcn.yaml
expts/hydra-configs/training/model/toymix_gin.yaml
expts/neurips2023_configs/config_classifigression_l1000.yaml
expts/neurips2023_configs/config_large_gcn.yaml
expts/neurips2023_configs/config_large_gcn_g25.yaml
expts/neurips2023_configs/config_large_gcn_gpu.yaml
expts/neurips2023_configs/config_large_gcn_n4.yaml
expts/neurips2023_configs/config_large_gcn_pcba.yaml
expts/neurips2023_configs/config_large_gin.yaml
expts/neurips2023_configs/config_large_gin_g25.yaml
expts/neurips2023_configs/config_large_gin_n4.yaml
expts/neurips2023_configs/config_large_gin_pcba.yaml
expts/neurips2023_configs/config_large_gine.yaml
expts/neurips2023_configs/config_large_gine_g25.yaml
expts/neurips2023_configs/config_large_gine_n4.yaml
expts/neurips2023_configs/config_large_gine_pcba.yaml
expts/neurips2023_configs/config_large_mpnn.yaml
expts/neurips2023_configs/config_luis_jama.yaml
expts/neurips2023_configs/config_small_gated_gcn.yaml
expts/neurips2023_configs/config_small_gcn.yaml
expts/neurips2023_configs/config_small_gcn_gpu.yaml
expts/neurips2023_configs/config_small_gin.yaml
expts/neurips2023_configs/config_small_gine.yaml
expts/neurips2023_configs/config_small_mpnn.yaml
expts/neurips2023_configs/base_config/large.yaml
expts/neurips2023_configs/base_config/large_pcba.yaml
expts/neurips2023_configs/base_config/large_pcqm_g25.yaml
expts/neurips2023_configs/base_config/large_pcqm_n4.yaml
expts/neurips2023_configs/base_config/small.yaml
expts/neurips2023_configs/baseline/config_small_gcn_baseline.yaml
expts/neurips2023_configs/baseline/config_small_gin_baseline.yaml
expts/neurips2023_configs/baseline/config_small_gine_baseline.yaml
expts/neurips2023_configs/debug/config_debug.yaml
expts/neurips2023_configs/debug/config_large_gcn_debug.yaml
expts/neurips2023_configs/debug/config_small_gcn_debug.yaml
expts/neurips2023_configs/single_task_gcn/config_large_gcn_mcf7.yaml
expts/neurips2023_configs/single_task_gcn/config_large_gcn_pcba.yaml
expts/neurips2023_configs/single_task_gcn/config_large_gcn_vcap.yaml
expts/neurips2023_configs/single_task_gin/config_large_gin_g25.yaml
expts/neurips2023_configs/single_task_gin/config_large_gin_mcf7.yaml
expts/neurips2023_configs/single_task_gin/config_large_gin_n4.yaml
expts/neurips2023_configs/single_task_gin/config_large_gin_pcba.yaml
expts/neurips2023_configs/single_task_gin/config_large_gin_pcq.yaml
expts/neurips2023_configs/single_task_gin/config_large_gin_vcap.yaml
expts/neurips2023_configs/single_task_gine/config_large_gine_g25.yaml
expts/neurips2023_configs/single_task_gine/config_large_gine_mcf7.yaml
expts/neurips2023_configs/single_task_gine/config_large_gine_n4.yaml
expts/neurips2023_configs/single_task_gine/config_large_gine_pcba.yaml
expts/neurips2023_configs/single_task_gine/config_large_gine_pcq.yaml
expts/neurips2023_configs/single_task_gine/config_large_gine_vcap.yaml
graphium/__init__.py
graphium/_version.py
graphium.egg-info/PKG-INFO
graphium.egg-info/SOURCES.txt
graphium.egg-info/dependency_links.txt
graphium.egg-info/entry_points.txt
graphium.egg-info/requires.txt
graphium.egg-info/top_level.txt
graphium/cli/__init__.py
graphium/cli/__main__.py
graphium/cli/data.py
graphium/cli/finetune_utils.py
graphium/cli/fingerprints.py
graphium/cli/main.py
graphium/cli/train_finetune_test.py
graphium/config/README.md
graphium/config/__init__.py
graphium/config/_load.py
graphium/config/_loader.py
graphium/config/config_convert.py
graphium/config/dummy_finetuning_from_gnn.yaml
graphium/config/dummy_finetuning_from_task_head.yaml
graphium/config/fake_and_missing_multilevel_multitask_pyg.yaml
graphium/config/fake_multilevel_multitask_pyg.yaml
graphium/config/zinc_default_multitask_pyg.yaml
graphium/data/README.md
graphium/data/__init__.py
graphium/data/collate.py
graphium/data/datamodule.py
graphium/data/dataset.py
graphium/data/multilevel_utils.py
graphium/data/normalization.py
graphium/data/sampler.py
graphium/data/sdf2csv.py
graphium/data/smiles_transform.py
graphium/data/utils.py
graphium/data/L1000/datasets_goli-L1000_parse_gctx_to_csv.ipynb
graphium/data/QM9/micro_qm9.csv
graphium/data/QM9/micro_qm9.parquet
graphium/data/QM9/norm_micro_qm9.csv
graphium/data/make_data_splits/make_train_val_test_splits_large.ipynb
graphium/data/make_data_splits/make_train_val_test_splits_small.ipynb
graphium/data/micro_ZINC/__init__.py
graphium/data/micro_ZINC/micro_ZINC.csv
graphium/data/multitask/__init__.py
graphium/data/multitask/tiny_ZINC_SA.csv
graphium/data/multitask/tiny_ZINC_logp.csv
graphium/data/multitask/tiny_ZINC_score.csv
graphium/data/single_atom_dataset/single_atom_dataset.csv
graphium/expts/pyg_batching_sparse.ipynb
graphium/features/README.md
graphium/features/__init__.py
graphium/features/commute.py
graphium/features/electrostatic.py
graphium/features/featurizer.py
graphium/features/graphormer.py
graphium/features/nmp.py
graphium/features/periodic_table.csv
graphium/features/positional_encoding.py
graphium/features/properties.py
graphium/features/rw.py
graphium/features/spectral.py
graphium/features/test_new_pes.ipynb
graphium/features/transfer_pos_level.py
graphium/finetuning/__init__.py
graphium/finetuning/finetuning.py
graphium/finetuning/finetuning_architecture.py
graphium/finetuning/fingerprinting.py
graphium/finetuning/utils.py
graphium/hyper_param_search/__init__.py
graphium/hyper_param_search/results.py
graphium/ipu/README.md
graphium/ipu/__init__.py
graphium/ipu/ipu_dataloader.py
graphium/ipu/ipu_losses.py
graphium/ipu/ipu_metrics.py
graphium/ipu/ipu_simple_lightning.py
graphium/ipu/ipu_utils.py
graphium/ipu/ipu_wrapper.py
graphium/ipu/to_dense_batch.py
graphium/nn/README.md
graphium/nn/__init__.py
graphium/nn/base_graph_layer.py
graphium/nn/base_layers.py
graphium/nn/residual_connections.py
graphium/nn/utils.py
graphium/nn/architectures/README.md
graphium/nn/architectures/__init__.py
graphium/nn/architectures/encoder_manager.py
graphium/nn/architectures/global_architectures.py
graphium/nn/architectures/pyg_architectures.py
graphium/nn/encoders/README.md
graphium/nn/encoders/__init__.py
graphium/nn/encoders/base_encoder.py
graphium/nn/encoders/bessel_pos_encoder.py
graphium/nn/encoders/gaussian_kernel_pos_encoder.py
graphium/nn/encoders/laplace_pos_encoder.py
graphium/nn/encoders/mlp_encoder.py
graphium/nn/encoders/signnet_pos_encoder.py
graphium/nn/pyg_layers/README.md
graphium/nn/pyg_layers/__init__.py
graphium/nn/pyg_layers/dimenet_pyg.py
graphium/nn/pyg_layers/gated_gcn_pyg.py
graphium/nn/pyg_layers/gcn_pyg.py
graphium/nn/pyg_layers/gin_pyg.py
graphium/nn/pyg_layers/gps_pyg.py
graphium/nn/pyg_layers/mpnn_pyg.py
graphium/nn/pyg_layers/pna_pyg.py
graphium/nn/pyg_layers/pooling_pyg.py
graphium/nn/pyg_layers/utils.py
graphium/trainer/README.md
graphium/trainer/__init__.py
graphium/trainer/losses.py
graphium/trainer/metrics.py
graphium/trainer/predictor.py
graphium/trainer/predictor_options.py
graphium/trainer/predictor_summaries.py
graphium/utils/README.md
graphium/utils/__init__.py
graphium/utils/arg_checker.py
graphium/utils/command_line_utils.py
graphium/utils/custom_lr.py
graphium/utils/decorators.py
graphium/utils/dict_tensor.py
graphium/utils/fs.py
graphium/utils/hashing.py
graphium/utils/moving_average_tracker.py
graphium/utils/mup.py
graphium/utils/packing.py
graphium/utils/read_file.py
graphium/utils/safe_run.py
graphium/utils/spaces.py
graphium/utils/tensor.py
graphium/visualization/README.md
graphium/visualization/__init__.py
graphium/visualization/vis_utils.py
notebooks/compare-pretraining-finetuning-performance.ipynb
notebooks/dev-datamodule-invalidate-cache.ipynb
notebooks/dev-datamodule-ogb.ipynb
notebooks/dev-datamodule.ipynb
notebooks/dev-pretrained.ipynb
notebooks/dev-training-loop.ipynb
notebooks/dev.ipynb
notebooks/finetuning-on-tdc-admet-benchmark.ipynb
notebooks/running-fingerprints-from-pretrained-model.ipynb
notebooks/running-model-from-config.ipynb
profiling/configs_profiling.yaml
profiling/profile_mol_to_graph.py
profiling/profile_one_of_k_encoding.py
profiling/profile_predictor.py
scripts/convert_yml.py
scripts/ipu_start.sh
scripts/ipu_venv.sh
tests/.gitignore
tests/__init__.py
tests/config_test_ipu_dataloader.yaml
tests/config_test_ipu_dataloader_multitask.yaml
tests/conftest.py
tests/converted_fake_multilevel_data.parquet
tests/dummy-pretrained-model.ckpt
tests/fake_and_missing_multilevel_data.parquet
tests/test_architectures.py
tests/test_attention.py
tests/test_base_layers.py
tests/test_collate.py
tests/test_data_utils.py
tests/test_datamodule.py
tests/test_dataset.py
tests/test_dict_tensor.py
tests/test_featurizer.py
tests/test_finetuning.py
tests/test_ipu_dataloader.py
tests/test_ipu_losses.py
tests/test_ipu_metrics.py
tests/test_ipu_options.py
tests/test_ipu_poptorch.py
tests/test_ipu_to_dense_batch.py
tests/test_ipu_training.py
tests/test_loaders.py
tests/test_losses.py
tests/test_metrics.py
tests/test_mtl_architecture.py
tests/test_multitask_datamodule.py
tests/test_mup.py
tests/test_packing.py
tests/test_pe_nodepair.py
tests/test_pe_rw.py
tests/test_pe_spectral.py
tests/test_pos_transfer_funcs.py
tests/test_positional_encoders.py
tests/test_positional_encodings.py
tests/test_predictor.py
tests/test_pyg_layers.py
tests/test_residual_connections.py
tests/test_utils.py
tests/data/config_micro_ZINC.yaml
tests/data/micro_ZINC.csv
tests/data/micro_ZINC_corrupt.csv
tests/data/micro_ZINC_shard_1.csv
tests/data/micro_ZINC_shard_1.parquet
tests/data/micro_ZINC_shard_2.csv
tests/data/micro_ZINC_shard_2.parquet
tests/data/pcqm4mv2-2k.csv