├── .gitattributes ├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py ├── config_large.yaml ├── config_medium.yaml ├── config_small.yaml └── test_config.yaml ├── docs ├── data.md ├── get_started.md ├── index.md ├── reference │ ├── data.md │ ├── models │ │ ├── graph.md │ │ ├── index.md │ │ └── transformer.md │ ├── observables.md │ ├── training.md │ └── utils.md └── resource │ ├── architecture.png │ └── architecture.tex ├── examples ├── 1_overview.ipynb ├── 2_Dataset.ipynb └── 3_Observables.ipynb ├── main.py ├── mkdocs.yaml ├── models ├── M_1 │ ├── checkpoints │ │ ├── best-checkpoint-epoch=30-train_loss=0.42.ckpt │ │ ├── best-checkpoint-epoch=41-train_loss=0.42.ckpt │ │ ├── best-checkpoint-epoch=50-train_loss=0.42.ckpt │ │ └── last.ckpt │ ├── hparams.yaml │ └── model_info.yaml ├── M_2 │ ├── checkpoints │ │ ├── best-checkpoint-epoch=29-train_loss=0.43.ckpt │ │ ├── best-checkpoint-epoch=4-train_loss=0.43.ckpt │ │ ├── best-checkpoint-epoch=5-train_loss=0.43.ckpt │ │ └── last.ckpt │ ├── hparams.yaml │ └── model_info.yaml └── M_3 │ ├── checkpoints │ ├── best-checkpoint-epoch=10-train_loss=0.44.ckpt │ ├── best-checkpoint-epoch=12-train_loss=0.44.ckpt │ ├── best-checkpoint-epoch=6-train_loss=0.45.ckpt │ └── last.ckpt │ ├── hparams.yaml │ └── model_info.yaml ├── pyproject.toml ├── setup.cfg ├── setup.py └── src └── rydberggpt ├── __init__.py ├── data ├── __init__.py ├── dataclasses.py ├── graph_structures.py ├── rydberg_dataset.py └── utils_graph.py ├── models ├── __init__.py ├── graph_embedding │ ├── __init__.py │ ├── layers.py │ └── models.py ├── rydberg_decoder_wavefunction.py ├── rydberg_encoder_decoder.py ├── transformer │ ├── __init__.py │ ├── layers.py │ ├── models.py │ ├── modules.py │ └── utils.py └── utils.py ├── observables ├── __init__.py └── rydberg_energy.py ├── tests ├── dataset_test │ ├── L_5 │ │ ├── BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0 │ │ │ ├── config.json │ │ │ ├── dataset.h5 │ │ │ └── graph.json │ │ └── BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0 │ │ │ ├── config.json │ │ │ ├── dataset.h5 │ │ │ └── graph.json │ └── L_6 │ │ ├── BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0 │ │ ├── config.json │ │ ├── dataset.h5 │ │ └── graph.json │ │ └── BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0 │ │ ├── config.json │ │ ├── dataset.h5 │ │ └── graph.json ├── test_dataloader.py ├── test_observables.py └── test_transformer.py ├── training ├── __init__.py ├── callbacks │ ├── __init__.py │ ├── module_info_callback.py │ └── stop_on_loss_threshold_callback.py ├── logger.py ├── loss.py ├── monitoring.py ├── train.py ├── trainer.py └── utils.py ├── utils.py └── utils_ckpt.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/.gitattributes -------------------------------------------------------------------------------- /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/config_large.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/config/config_large.yaml -------------------------------------------------------------------------------- /config/config_medium.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/config/config_medium.yaml -------------------------------------------------------------------------------- /config/config_small.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/config/config_small.yaml -------------------------------------------------------------------------------- /config/test_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/config/test_config.yaml -------------------------------------------------------------------------------- /docs/data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/data.md -------------------------------------------------------------------------------- /docs/get_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/get_started.md -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/index.md -------------------------------------------------------------------------------- /docs/reference/data.md: -------------------------------------------------------------------------------- 1 | ::: rydberggpt.data 2 | -------------------------------------------------------------------------------- /docs/reference/models/graph.md: -------------------------------------------------------------------------------- 1 | ::: rydberggpt.models.graph_embedding -------------------------------------------------------------------------------- /docs/reference/models/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/reference/models/index.md -------------------------------------------------------------------------------- /docs/reference/models/transformer.md: -------------------------------------------------------------------------------- 1 | ::: rydberggpt.models.transformer -------------------------------------------------------------------------------- /docs/reference/observables.md: -------------------------------------------------------------------------------- 1 | ::: rydberggpt.observables 2 | -------------------------------------------------------------------------------- /docs/reference/training.md: -------------------------------------------------------------------------------- 1 | ::: rydberggpt.training -------------------------------------------------------------------------------- /docs/reference/utils.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/reference/utils.md -------------------------------------------------------------------------------- /docs/resource/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/resource/architecture.png -------------------------------------------------------------------------------- /docs/resource/architecture.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/docs/resource/architecture.tex -------------------------------------------------------------------------------- /examples/1_overview.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/examples/1_overview.ipynb -------------------------------------------------------------------------------- /examples/2_Dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/examples/2_Dataset.ipynb -------------------------------------------------------------------------------- /examples/3_Observables.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/examples/3_Observables.ipynb -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/main.py -------------------------------------------------------------------------------- /mkdocs.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/mkdocs.yaml -------------------------------------------------------------------------------- /models/M_1/checkpoints/best-checkpoint-epoch=30-train_loss=0.42.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_1/checkpoints/best-checkpoint-epoch=30-train_loss=0.42.ckpt -------------------------------------------------------------------------------- /models/M_1/checkpoints/best-checkpoint-epoch=41-train_loss=0.42.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_1/checkpoints/best-checkpoint-epoch=41-train_loss=0.42.ckpt -------------------------------------------------------------------------------- /models/M_1/checkpoints/best-checkpoint-epoch=50-train_loss=0.42.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_1/checkpoints/best-checkpoint-epoch=50-train_loss=0.42.ckpt -------------------------------------------------------------------------------- /models/M_1/checkpoints/last.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_1/checkpoints/last.ckpt -------------------------------------------------------------------------------- /models/M_1/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_1/hparams.yaml -------------------------------------------------------------------------------- /models/M_1/model_info.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_1/model_info.yaml -------------------------------------------------------------------------------- /models/M_2/checkpoints/best-checkpoint-epoch=29-train_loss=0.43.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_2/checkpoints/best-checkpoint-epoch=29-train_loss=0.43.ckpt -------------------------------------------------------------------------------- /models/M_2/checkpoints/best-checkpoint-epoch=4-train_loss=0.43.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_2/checkpoints/best-checkpoint-epoch=4-train_loss=0.43.ckpt -------------------------------------------------------------------------------- /models/M_2/checkpoints/best-checkpoint-epoch=5-train_loss=0.43.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_2/checkpoints/best-checkpoint-epoch=5-train_loss=0.43.ckpt -------------------------------------------------------------------------------- /models/M_2/checkpoints/last.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_2/checkpoints/last.ckpt -------------------------------------------------------------------------------- /models/M_2/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_2/hparams.yaml -------------------------------------------------------------------------------- /models/M_2/model_info.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_2/model_info.yaml -------------------------------------------------------------------------------- /models/M_3/checkpoints/best-checkpoint-epoch=10-train_loss=0.44.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_3/checkpoints/best-checkpoint-epoch=10-train_loss=0.44.ckpt -------------------------------------------------------------------------------- /models/M_3/checkpoints/best-checkpoint-epoch=12-train_loss=0.44.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_3/checkpoints/best-checkpoint-epoch=12-train_loss=0.44.ckpt -------------------------------------------------------------------------------- /models/M_3/checkpoints/best-checkpoint-epoch=6-train_loss=0.45.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_3/checkpoints/best-checkpoint-epoch=6-train_loss=0.45.ckpt -------------------------------------------------------------------------------- /models/M_3/checkpoints/last.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_3/checkpoints/last.ckpt -------------------------------------------------------------------------------- /models/M_3/hparams.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_3/hparams.yaml -------------------------------------------------------------------------------- /models/M_3/model_info.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/models/M_3/model_info.yaml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/setup.py -------------------------------------------------------------------------------- /src/rydberggpt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/data/dataclasses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/data/dataclasses.py -------------------------------------------------------------------------------- /src/rydberggpt/data/graph_structures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/data/graph_structures.py -------------------------------------------------------------------------------- /src/rydberggpt/data/rydberg_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/data/rydberg_dataset.py -------------------------------------------------------------------------------- /src/rydberggpt/data/utils_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/data/utils_graph.py -------------------------------------------------------------------------------- /src/rydberggpt/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/models/graph_embedding/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/models/graph_embedding/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/graph_embedding/layers.py -------------------------------------------------------------------------------- /src/rydberggpt/models/graph_embedding/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/graph_embedding/models.py -------------------------------------------------------------------------------- /src/rydberggpt/models/rydberg_decoder_wavefunction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/rydberg_decoder_wavefunction.py -------------------------------------------------------------------------------- /src/rydberggpt/models/rydberg_encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/rydberg_encoder_decoder.py -------------------------------------------------------------------------------- /src/rydberggpt/models/transformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/models/transformer/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/transformer/layers.py -------------------------------------------------------------------------------- /src/rydberggpt/models/transformer/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/transformer/models.py -------------------------------------------------------------------------------- /src/rydberggpt/models/transformer/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/transformer/modules.py -------------------------------------------------------------------------------- /src/rydberggpt/models/transformer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/transformer/utils.py -------------------------------------------------------------------------------- /src/rydberggpt/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/models/utils.py -------------------------------------------------------------------------------- /src/rydberggpt/observables/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/observables/rydberg_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/observables/rydberg_energy.py -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0/config.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0/dataset.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0/dataset.h5 -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0/graph.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=1.0/graph.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0/config.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0/dataset.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0/dataset.h5 -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0/graph.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_5/BloqadeQMC_L=5_Rb=1.05_delta=-0.13_beta=2.0/graph.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0/config.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0/dataset.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0/dataset.h5 -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0/graph.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=1.0/graph.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0/config.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0/dataset.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0/dataset.h5 -------------------------------------------------------------------------------- /src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0/graph.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/dataset_test/L_6/BloqadeQMC_L=6_Rb=1.05_delta=-0.13_beta=2.0/graph.json -------------------------------------------------------------------------------- /src/rydberggpt/tests/test_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/test_dataloader.py -------------------------------------------------------------------------------- /src/rydberggpt/tests/test_observables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/test_observables.py -------------------------------------------------------------------------------- /src/rydberggpt/tests/test_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/tests/test_transformer.py -------------------------------------------------------------------------------- /src/rydberggpt/training/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/training/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/rydberggpt/training/callbacks/module_info_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/callbacks/module_info_callback.py -------------------------------------------------------------------------------- /src/rydberggpt/training/callbacks/stop_on_loss_threshold_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/callbacks/stop_on_loss_threshold_callback.py -------------------------------------------------------------------------------- /src/rydberggpt/training/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/logger.py -------------------------------------------------------------------------------- /src/rydberggpt/training/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/loss.py -------------------------------------------------------------------------------- /src/rydberggpt/training/monitoring.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/monitoring.py -------------------------------------------------------------------------------- /src/rydberggpt/training/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/train.py -------------------------------------------------------------------------------- /src/rydberggpt/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/trainer.py -------------------------------------------------------------------------------- /src/rydberggpt/training/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/training/utils.py -------------------------------------------------------------------------------- /src/rydberggpt/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/utils.py -------------------------------------------------------------------------------- /src/rydberggpt/utils_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PIQuIL/RydbergGPT/HEAD/src/rydberggpt/utils_ckpt.py --------------------------------------------------------------------------------