sagemaker>=2.145.0
boto3>=1.34.153
botocore>=1.34.153
psutil

[cdk]
aws-cdk-lib

[dev]
sagemaker-pytorch-training
sagemaker-pytorch-inference
torch-model-archiver
tox
wheel
build
twine
pydevd-pycharm~=222.4459.20
scikit-learn
transformers
py-cpuinfo
deepspeed

[dev-macos]
tensorflow-macos==2.9.2
numpy==1.22.4

[test]
build
coverage
flake8
mock
pydocstyle
pytest
pytest-cov
pytest-html
pytest-profiling
pytest-monitor
bandit
aws-cdk-lib
sagemaker-studio-image-build
sagemaker-training
selenium
accelerate[sagemaker]<0.29
