LICENSE
README.md
pyproject.toml
setup.cfg
src/itwinai/__init__.py
src/itwinai/cli.py
src/itwinai/components.py
src/itwinai/constants.py
src/itwinai/distributed.py
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_constants.py
src/itwinai/slurm/slurm_script_builder.py
src/itwinai/slurm/slurm_script_configuration.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/distributed.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/utils.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/test_cli.py
tests/test_slurm_builder.py
tests/test_utils.py