.dockerignore
.gitignore
.gitmodules
.python-version
.readthedocs.yaml
CHANGELOG
CITATION.cff
CODEOWNERS
CONTRIBUTING.md
COPYRIGHT
LICENSE
MAINTAINERS.md
Makefile
README.md
THIRD_PARTY_LICENSES
pyproject.toml
setup.cfg
uv.lock
.github/CODE_OF_CONDUCT.md
.github/ISSUE_TEMPLATE.md
.github/PULL_REQUEST_TEMPLATE.md
.github/dependabot.yml
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/actions/build-container/action.yml
.github/linters/.flake8
.github/linters/.hadolint.yaml
.github/linters/.isort.cfg
.github/linters/.jscpd.json
.github/linters/.markdownlint.json
.github/linters/.ruff.toml
.github/linters/mlc_config.json
.github/workflows/check-links.yml
.github/workflows/container-ci.yml
.github/workflows/lint.yml
.github/workflows/pypi.yml
.github/workflows/pytest.yml
.github/workflows/sqaaas.yml
.vscode/extensions.json
.vscode/settings.json
ci/.gitattributes
ci/.gitignore
ci/dagger.json
ci/examples.sh
ci/pyproject.toml
ci/uv.lock
ci/src/main/__init__.py
ci/src/main/hpc.py
ci/src/main/interlink.py
ci/src/main/k8s.py
ci/src/main/literals.py
ci/src/main/main.py
ci/src/main/utils.py
docs/Makefile
docs/README.md
docs/conf.py
docs/convert_admonitions.py
docs/index.rst
docs/make.bat
docs/requirements.txt
docs/testing-with-pytest.md
docs/working-with-containers.md
docs/_static/custom.css
docs/api/cli.md
docs/api/cli_reference.rst
docs/api/itwinai.components.rst
docs/api/itwinai.distributed.rst
docs/api/itwinai.loggers.rst
docs/api/itwinai.parser.rst
docs/api/itwinai.pipeline.rst
docs/api/itwinai.scalability_report.modules.rst
docs/api/itwinai.serialization.rst
docs/api/itwinai.tests.modules.rst
docs/api/itwinai.tf.modules.rst
docs/api/itwinai.torch.modules.rst
docs/api/itwinai.type.rst
docs/api/itwinai.utils.rst
docs/api/modules.rst
docs/getting-started/containers.rst
docs/getting-started/getting_started_with_itwinai.rst
docs/getting-started/plugins-list.rst
docs/getting-started/plugins.rst
docs/getting-started/slurm.rst
docs/how-it-works/hpo/explain-hpo.rst
docs/how-it-works/hpo/images/hpo_fig_eurac.svg
docs/how-it-works/loggers/explain_loggers.rst
docs/how-it-works/loggers/figures/logger_fig.png
docs/how-it-works/training/explain_ddp.rst
docs/how-it-works/training/training.rst
docs/how-it-works/workflows/explain_workflows.rst
docs/how-it-works/workflows/figures/Adapt_example.png
docs/how-it-works/workflows/figures/Advanced_workflow.png
docs/how-it-works/workflows/figures/comp_Adapt.png
docs/how-it-works/workflows/figures/comp_Get.png
docs/how-it-works/workflows/figures/comp_Predict.png
docs/how-it-works/workflows/figures/comp_Proc.png
docs/how-it-works/workflows/figures/comp_Save.png
docs/how-it-works/workflows/figures/comp_Split.png
docs/how-it-works/workflows/figures/comp_Train.png
docs/how-it-works/workflows/figures/simple_pipeline.png
docs/images/icon-itwinai-orange-black-subtitle-small.png
docs/images/icon-itwinai-orange-black-subtitle.png
docs/images/icon-itwinai-orange-white-subtitle.png
docs/images/icon-itwinai-orange-white.png
docs/images/icon-itwinai-orange.png
docs/images/icon-itwinai-white.png
docs/installation/creating_venv.rst
docs/installation/developer_installation.rst
docs/installation/horovod_deepspeed_installation.rst
docs/installation/hpc_modules.rst
docs/installation/user_installation.rst
docs/installation/uv_tutorial.rst
docs/tutorials/tutorials.rst
docs/tutorials/distrib-ml/kuberay-setup-tutorial.rst
docs/tutorials/distrib-ml/tf_scaling_test.rst
docs/tutorials/distrib-ml/tf_tutorial_0_basics.rst
docs/tutorials/distrib-ml/tf_tutorial_1_imagenet.rst
docs/tutorials/distrib-ml/torch-tutorial-GAN.rst
docs/tutorials/distrib-ml/torch-tutorial-containers.rst
docs/tutorials/distrib-ml/torch_scaling_test.rst
docs/tutorials/distrib-ml/torch_tutorial_0_basics.rst
docs/tutorials/distrib-ml/torch_tutorial_1_mnist.rst
docs/tutorials/distrib-ml/torch_tutorial_2_trainer_class.rst
docs/tutorials/distrib-ml/torch_tutorial_kubeflow_1.rst
docs/tutorials/hpo-workflows/hpo-torchtrainer-integration.rst
docs/tutorials/workflows/01-pipeline-introduction
docs/tutorials/workflows/02-pipeline-configuration
docs/tutorials/workflows/03-dag-workflows
docs/tutorials/workflows/04_itwinai_argparser.rst
docs/use-cases/3dgan_doc.rst
docs/use-cases/cyclones_doc.rst
docs/use-cases/eurac_doc.rst
docs/use-cases/latticeqcd_doc.rst
docs/use-cases/mnist_doc.rst
docs/use-cases/radio-astronomy.rst
docs/use-cases/use_cases.rst
docs/use-cases/virgo_doc.rst
docs/use-cases/xtclim_doc.rst
env-files/docs/build-docs-jsc.sh
env-files/docs/create-docs-env-jsc.sh
env-files/tensorflow/Dockerfile
env-files/tensorflow/createEnvJSCTF.sh
env-files/tensorflow/createEnvVegaTF.sh
env-files/tensorflow/generic_tf.sh
env-files/torch/Dockerfile
env-files/torch/README.md
env-files/torch/createEnvJSC.sh
env-files/torch/createEnvVega.sh
env-files/torch/generic_torch.sh
env-files/torch/horovod-deepspeed-JSC.slurm
env-files/torch/install-horovod-deepspeed-cuda.sh
env-files/torch/rocm.def
env-files/torch/skinny.Dockerfile
env-files/torch/slim-v1.Dockerfile
env-files/torch/slim.Dockerfile
env-files/torch/jupyter/Dockerfile
env-files/torch/jupyter/README.md
env-files/torch/jupyter/asyncssh_config.py
env-files/torch/jupyter/configure.py
env-files/torch/jupyter/ipython_kernel_config.json
env-files/torch/jupyter/old.slim.Dockerfile
env-files/torch/jupyter/setup.sh
env-files/torch/jupyter/slim.Dockerfile
env-files/torch/requirements/README.md
env-files/torch/requirements/cmcc-requirements.txt
env-files/torch/requirements/requirements.txt
licenses/CDDL-1.0.txt
src/itwinai/__init__.py
src/itwinai/cli.py
src/itwinai/components.py
src/itwinai/distributed.py
src/itwinai/flamegraph.pl
src/itwinai/loggers.py
src/itwinai/parser.py
src/itwinai/pipeline.py
src/itwinai/serialization.py
src/itwinai/type.py
src/itwinai/utils.py
src/itwinai.egg-info/PKG-INFO
src/itwinai.egg-info/SOURCES.txt
src/itwinai.egg-info/dependency_links.txt
src/itwinai.egg-info/entry_points.txt
src/itwinai.egg-info/requires.txt
src/itwinai.egg-info/top_level.txt
src/itwinai/plugins/__init__.py
src/itwinai/scalability_report/data.py
src/itwinai/scalability_report/plot.py
src/itwinai/scalability_report/reports.py
src/itwinai/scalability_report/utils.py
src/itwinai/slurm/slurm_config.yaml
src/itwinai/slurm/slurm_constants.py
src/itwinai/slurm/slurm_script_builder.py
src/itwinai/slurm/utils.py
src/itwinai/tensorflow/__init__.py
src/itwinai/tensorflow/distributed.py
src/itwinai/tensorflow/trainer.py
src/itwinai/tensorflow/utils.py
src/itwinai/tensorflow/data/__init__.py
src/itwinai/tensorflow/models/__init__.py
src/itwinai/tensorflow/models/mnist.py
src/itwinai/tests/__init__.py
src/itwinai/tests/dummy_components.py
src/itwinai/tests/exceptions.py
src/itwinai/tests/sanity_check.py
src/itwinai/torch/__init__.py
src/itwinai/torch/config.py
src/itwinai/torch/distributed.py
src/itwinai/torch/gan.py
src/itwinai/torch/inference.py
src/itwinai/torch/loggers.py
src/itwinai/torch/mlflow.py
src/itwinai/torch/reproducibility.py
src/itwinai/torch/trainer.py
src/itwinai/torch/tuning.py
src/itwinai/torch/type.py
src/itwinai/torch/data/__init__.py
src/itwinai/torch/models/__init__.py
src/itwinai/torch/models/mnist.py
src/itwinai/torch/monitoring/backend.py
src/itwinai/torch/monitoring/monitoring.py
src/itwinai/torch/profiling/profiler.py
src/itwinai/torch/profiling/py_spy_aggregation.py
tests/conftest.py
tests/run_on_jsc.sh
tests/test_cli.py
tests/test_slurm_builder.py
tests/test_utils.py
tests/components/conftest.py
tests/components/test_components.py
tests/components/test_config_parsing.py
tests/components/test_decorators.py
tests/components/test_pipeline.py
tests/loggers/conftest.py
tests/loggers/test_lightning_logger.py
tests/loggers/test_loggers.py
tests/loggers/test_prov4ml.py
tests/torch/README.md
tests/torch/conftest.py
tests/torch/distributed_decorator.py
tests/torch/runall.jsc.sh
tests/torch/runall.vega.sh
tests/torch/slurm.jsc.sh
tests/torch/slurm.vega.sh
tests/torch/test_checkpointing.py
tests/torch/test_config.py
tests/torch/test_distributed.py
tests/torch/test_ray_search_space.py
tests/torch/test_torch_trainer.py
tests/torch/test_tuning.py
tests/use-cases/conftest.py
tests/use-cases/test_3dgan.py
tests/use-cases/test_cyclones.py
tests/use-cases/test_mnist.py
tests/use-cases/test_radio_astronomy.py
tutorials/distributed-ml/kuberay-setup-tutorial/README.md
tutorials/distributed-ml/kuberay-setup-tutorial/raycluster_example.yaml
tutorials/distributed-ml/tf-scaling-test-jube/README.md
tutorials/distributed-ml/tf-scaling-test-jube/bench_plot.ipynb
tutorials/distributed-ml/tf-scaling-test-jube/general_jobsys.xml
tutorials/distributed-ml/tf-scaling-test-jube/jube_ddp.sh
tutorials/distributed-ml/tf-scaling-test-jube/train.py
tutorials/distributed-ml/tf-tutorial-0-basics/README.md
tutorials/distributed-ml/tf-tutorial-0-basics/tfmirrored_slurm.sh
tutorials/distributed-ml/tf-tutorial-0-basics/train.py
tutorials/distributed-ml/tf-tutorial-1-imagenet/README.md
tutorials/distributed-ml/tf-tutorial-1-imagenet/tfmirrored_slurm.sh
tutorials/distributed-ml/tf-tutorial-1-imagenet/train.py
tutorials/distributed-ml/torch-kubeflow-1/Dockerfile
tutorials/distributed-ml/torch-kubeflow-1/README.md
tutorials/distributed-ml/torch-kubeflow-1/cpu.yaml
tutorials/distributed-ml/torch-kubeflow-1/train-cpu.py
tutorials/distributed-ml/torch-scaling-test/README.md
tutorials/distributed-ml/torch-scaling-test/ddp_trainer.py
tutorials/distributed-ml/torch-scaling-test/deepspeed_trainer.py
tutorials/distributed-ml/torch-scaling-test/horovod_trainer.py
tutorials/distributed-ml/torch-scaling-test/itwinai_trainer.py
tutorials/distributed-ml/torch-scaling-test/slurm.py
tutorials/distributed-ml/torch-scaling-test/slurm_config.yaml
tutorials/distributed-ml/torch-scaling-test/utils.py
tutorials/distributed-ml/torch-scaling-test/config/base.yaml
tutorials/distributed-ml/torch-scaling-test/config/ddp.yaml
tutorials/distributed-ml/torch-scaling-test/config/deepspeed.yaml
tutorials/distributed-ml/torch-scaling-test/config/horovod.yaml
tutorials/distributed-ml/torch-scaling-test/img/report.png
tutorials/distributed-ml/torch-tutorial-0-basics/README.md
tutorials/distributed-ml/torch-tutorial-0-basics/slurm_config.yaml
tutorials/distributed-ml/torch-tutorial-0-basics/train.py
tutorials/distributed-ml/torch-tutorial-1-mnist/.gitignore
tutorials/distributed-ml/torch-tutorial-1-mnist/README.md
tutorials/distributed-ml/torch-tutorial-1-mnist/config.yaml
tutorials/distributed-ml/torch-tutorial-1-mnist/slurm_config.yaml
tutorials/distributed-ml/torch-tutorial-1-mnist/train.py
tutorials/distributed-ml/torch-tutorial-2-trainer-class/README.md
tutorials/distributed-ml/torch-tutorial-2-trainer-class/sample_code.py
tutorials/distributed-ml/torch-tutorial-2-trainer-class/sample_srun.sh
tutorials/distributed-ml/torch-tutorial-2-trainer-class/slurm_config.yaml
tutorials/distributed-ml/torch-tutorial-2-trainer-class/train.py
tutorials/distributed-ml/torch-tutorial-GAN/README.md
tutorials/distributed-ml/torch-tutorial-GAN/slurm_config.yaml
tutorials/distributed-ml/torch-tutorial-GAN/train.py
tutorials/distributed-ml/torch-tutorial-containers/README.md
tutorials/distributed-ml/torch-tutorial-containers/config.yaml
tutorials/distributed-ml/torch-tutorial-containers/dataloader.py
tutorials/distributed-ml/torch-tutorial-containers/model.py
tutorials/distributed-ml/torch-tutorial-containers/run_docker.sh
tutorials/distributed-ml/torch-tutorial-containers/runall.sh
tutorials/distributed-ml/torch-tutorial-containers/slurm.sh
tutorials/hpo-workflows/fashion-mnist/config.yaml
tutorials/hpo-workflows/fashion-mnist/data.py
tutorials/hpo-workflows/fashion-mnist/slurm_hpo.sh
tutorials/hpo-workflows/fashion-mnist/trainer.py
tutorials/ml-workflows/.gitignore
tutorials/ml-workflows/01-pipeline-introduction/basic_components.py
tutorials/ml-workflows/01-pipeline-introduction/sample_pipeline_1.jpg
tutorials/ml-workflows/01-pipeline-introduction/sample_pipeline_2.jpg
tutorials/ml-workflows/01-pipeline-introduction/tutorial_0_basic_workflow.ipynb
tutorials/ml-workflows/02-pipeline-configuration/tutorial_1_intermediate_workflow.rst
tutorials/ml-workflows/03-dag-workflows/Advanced_workflow.png
tutorials/ml-workflows/03-dag-workflows/basic_components.py
tutorials/ml-workflows/03-dag-workflows/tutorial_2_advanced_workflow.ipynb
tutorials/ml-workflows/04-itwinai-argparser/README.md
tutorials/ml-workflows/04-itwinai-argparser/basic_components.py
tutorials/ml-workflows/04-itwinai-argparser/main.py
use-cases/README.md
use-cases/3dgan/Dockerfile
use-cases/3dgan/README.md
use-cases/3dgan/config.yaml
use-cases/3dgan/create_inference_sample.py
use-cases/3dgan/dataloader.py
use-cases/3dgan/downsample_h5py_file.py
use-cases/3dgan/model.py
use-cases/3dgan/requirements.txt
use-cases/3dgan/run-provenance-experiments.sh
use-cases/3dgan/saver.py
use-cases/3dgan/slurm.jsc.sh
use-cases/3dgan/slurm.vega.sh
use-cases/3dgan/trainer.py
use-cases/3dgan/interLink/3dgan-inference-cpu.yaml
use-cases/3dgan/interLink/3dgan-inference.yaml
use-cases/3dgan/interLink/3dgan-train.yaml
use-cases/3dgan/interLink/README.md
use-cases/cyclones/.gitignore
use-cases/cyclones/README.md
use-cases/cyclones/cyclones_vgg.py
use-cases/cyclones/dataloader.py
use-cases/cyclones/pipeline.yaml
use-cases/cyclones/requirements.txt
use-cases/cyclones/startscript.sh
use-cases/cyclones/train.py
use-cases/cyclones/trainer.py
use-cases/cyclones/src/callbacks.py
use-cases/cyclones/src/macros.py
use-cases/cyclones/src/scaling.py
use-cases/cyclones/src/strategy.py
use-cases/cyclones/src/transform.py
use-cases/cyclones/src/utils.py
use-cases/cyclones/src/tfrecords/dataset.py
use-cases/cyclones/src/tfrecords/functions.py
use-cases/eurac/.gitignore
use-cases/eurac/README.md
use-cases/eurac/config.yaml
use-cases/eurac/config_calibration.yaml
use-cases/eurac/config_training.yaml
use-cases/eurac/data.py
use-cases/eurac/hpo.py
use-cases/eurac/requirements.txt
use-cases/eurac/slurm_config.yaml
use-cases/eurac/slurm_ray.sh
use-cases/eurac/trainer.py
use-cases/eurac/scalability-plots/absolute_scalability_plot.png
use-cases/eurac/scalability-plots/communication_plot.png
use-cases/eurac/scalability-plots/gpu_energy_plot.png
use-cases/eurac/scalability-plots/relative_scalability_plot.png
use-cases/eurac/scalability-plots/utilization_plot.png
use-cases/lattice-qcd/README.md
use-cases/lattice-qcd/_normflowcore.py
use-cases/lattice-qcd/config.yaml
use-cases/lattice-qcd/setup.py
use-cases/lattice-qcd/startscript.sh
use-cases/lattice-qcd/train.py
use-cases/mnist/tensorflow/README.md
use-cases/mnist/tensorflow/dataloader.py
use-cases/mnist/tensorflow/pipeline.yaml
use-cases/mnist/tensorflow/startscript.sh
use-cases/mnist/torch/Dockerfile
use-cases/mnist/torch/README.md
use-cases/mnist/torch/config.yaml
use-cases/mnist/torch/create_inference_sample.py
use-cases/mnist/torch/dataloader.py
use-cases/mnist/torch/model.py
use-cases/mnist/torch/runall.jsc.sh
use-cases/mnist/torch/runall.lumi.sh
use-cases/mnist/torch/saver.py
use-cases/mnist/torch/slurm.jsc.sh
use-cases/mnist/torch/slurm.lumi.sh
use-cases/mnist/torch/slurm_config.yaml
use-cases/mnist/torch-lightning/README.md
use-cases/mnist/torch-lightning/config.yaml
use-cases/mnist/torch-lightning/dataloader.py
use-cases/mnist/torch-lightning/startscript
use-cases/mnist/torch-lightning/utils.py
use-cases/radio-astronomy/.config-test.yaml
use-cases/radio-astronomy/.gitignore
use-cases/radio-astronomy/.pytest-clean
use-cases/radio-astronomy/batch-jsc.sh
use-cases/radio-astronomy/clean
use-cases/radio-astronomy/config.yaml
use-cases/radio-astronomy/data.py
use-cases/radio-astronomy/inputs.py
use-cases/radio-astronomy/main.py
use-cases/radio-astronomy/requirements.txt
use-cases/radio-astronomy/trainer.py
use-cases/virgo/.gitignore
use-cases/virgo/README.md
use-cases/virgo/config.yaml
use-cases/virgo/data.py
use-cases/virgo/hpo.py
use-cases/virgo/requirements.txt
use-cases/virgo/slurm_config.yaml
use-cases/virgo/slurm_ray.sh
use-cases/virgo/trainer.py
use-cases/virgo/scalability-plots/absolute_scalability_plot.png
use-cases/virgo/scalability-plots/communication_plot.png
use-cases/virgo/scalability-plots/gpu_energy_plot.png
use-cases/virgo/scalability-plots/relative_scalability_plot.png
use-cases/virgo/scalability-plots/utilization_plot.png
use-cases/virgo/src/dataset.py
use-cases/virgo/src/model.py
use-cases/virgo/src/utils.py
use-cases/virgo/synthetic-data-gen/concat_hdf5_dataset_files.py
use-cases/virgo/synthetic-data-gen/data_generation_hdf5.sh
use-cases/virgo/synthetic-data-gen/file_gen_hdf5.py
use-cases/xtclim/LICENSE
use-cases/xtclim/README.md
use-cases/xtclim/hpo.py
use-cases/xtclim/pipeline.yaml
use-cases/xtclim/requirements.txt
use-cases/xtclim/slurm_ray.sh
use-cases/xtclim/startscript
use-cases/xtclim/train.py
use-cases/xtclim/outputs/.DS_Store
use-cases/xtclim/preprocessing/.DS_Store
use-cases/xtclim/preprocessing/preprocess_2d_seasons.py
use-cases/xtclim/preprocessing/preprocess_3d_seasons.py
use-cases/xtclim/preprocessing/preprocess_functions_2d_ssp.py
use-cases/xtclim/src/.DS_Store
use-cases/xtclim/src/anomaly.py
use-cases/xtclim/src/engine.py
use-cases/xtclim/src/initialization.py
use-cases/xtclim/src/model.py
use-cases/xtclim/src/scenario_season_comparison.ipynb
use-cases/xtclim/src/trainer.py
use-cases/xtclim/src/utils.py