프로덕션 머신러닝의 배포, 모니터링, 버전 관리, 확장 및 보안을 위한 오픈소스 라이브러리 모음입니다.
aws/sagemaker-python-sdk
Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 Python SDK로, 프레임워크와 내장 알고리즘, BYO 컨테이너를 지원합니다.
- 스타
- 2,270
- 포크
- 1,317
- 열린 이슈
- 237
- 마지막 커밋
- 어제
어떤 프로젝트인가요?
SageMaker Python SDK는 Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 오픈 소스 라이브러리입니다. 개발자는 PyTorch와 MXNet 같은 딥러닝 프레임워크, Amazon의 내장 알고리즘 또는 SageMaker 호환 Docker 컨테이너로 패키징한 사용자 지정 알고리즘을 사용해 Python 코드에서 학습 작업을 실행하고 모델을 호스팅할 수 있습니다.
3.x 버전은 학습용 ModelTrainer와 배포 및 추론용 ModelBuilder라는 통합 클래스를 중심으로 구성되며, 2.x의 프레임워크별 Estimator 및 Model 클래스를 대체합니다. 3.x는 core, training, serving, MLOps용 PyPI 패키지로 분리되어 있습니다. 모델 미세 조정 트레이너(SFT, DPO, RLAIF, RLVR)도 3.1 이상 버전에 포함되어 있습니다.
SDK는 Python 3.10~3.12를 사용하는 Unix/Linux와 macOS를 지원합니다. 사용하려면 AWS 계정과 SageMaker 권한이 있는 IAM 역할이 필요합니다.
핵심 기능
- 01SageMaker에서 PyTorch, MXNet 및 기타 프레임워크로 모델을 학습합니다.
- 02Amazon 내장 알고리즘을 사용하거나 자체 Docker 컨테이너를 가져와 사용할 수 있습니다.
- 03ModelBuilder로 학습된 모델을 실시간 엔드포인트에 배포합니다.
- 04LoRA 또는 전체 미세 조정을 사용해 파운데이션 모델을 미세 조정합니다.
- 05core, training, serving, MLOps용 모듈식 패키지를 설치합니다.
활용 사례
- Python에서 관리형 AWS GPU 인스턴스로 분산 학습 작업을 실행합니다.
- 학습된 모델을 실시간 추론 엔드포인트로 배포합니다.
- AWS에서 사용자 지정 데이터로 대규모 언어 모델을 미세 조정합니다.
- 모델 레지스트리와 처리 작업을 활용해 MLOps 파이프라인을 구축합니다.
빠른 시작
- 1
pip으로 PyPI에서 SDK를 설치합니다
pip install sagemaker
사용 전 알아둘 점
- Amazon SageMaker를 대상으로 하므로 AWS 계정과 IAM 권한이 필요합니다.
- README에 따르면 Unix/Linux와 Mac만 지원합니다.
- 버전 3.x에서는 버전 2의 Estimator, Model, Predictor 클래스가 제거되었습니다.
토픽
자주 묻는 질문
- sagemaker-python-sdk은(는) 무엇인가요?
- Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 Python SDK로, 프레임워크와 내장 알고리즘, BYO 컨테이너를 지원합니다. SageMaker Python SDK는 Amazon SageMaker에서 머신러닝 모델을 학습하고 배포하기 위한 오픈 소스 라이브러리입니다. 개발자는 PyTorch와 MXNet 같은 딥러닝 프레임워크, Amazon의 내장 알고리즘 또는 SageMaker 호환 Docker 컨테이너로 패키징한 사용자 지정 알고리즘을 사용해 Python 코드에서 학습 작업을 실행하고 모델을 호스팅할 수 있습니다.
- sagemaker-python-sdk은(는) 무료로 사용할 수 있나요? 라이선스는 무엇인가요?
- sagemaker-python-sdk은(는) Apache-2.0 라이선스의 오픈소스입니다. 상업용 제품에 사용하기 전에 라이선스 조건을 확인하세요.
- sagemaker-python-sdk은(는) 아직 유지보수되고 있나요?
- 활발히 유지보수 중. sagemaker-python-sdk의 마지막 커밋은 2026년 10월 23일입니다.
- sagemaker-python-sdk은(는) 어디에 사용할 수 있나요?
- Python에서 관리형 AWS GPU 인스턴스로 분산 학습 작업을 실행합니다. 학습된 모델을 실시간 추론 엔드포인트로 배포합니다. AWS에서 사용자 지정 데이터로 대규모 언어 모델을 미세 조정합니다. 모델 레지스트리와 처리 작업을 활용해 MLOps 파이프라인을 구축합니다.
- sagemaker-python-sdk는 어떻게 설치하나요?
- pip으로 PyPI에서 SDK를 설치합니다: pip install sagemaker
- sagemaker-python-sdk은(는) 어디서 받을 수 있나요?
- 소스 코드와 설치 방법은 GitHub에 있습니다: https://github.com/aws/sagemaker-python-sdk
요약은 프로젝트 README를 바탕으로 AI가 작성했습니다. GitHub 통계는 매일 갱신됩니다.
관련 프로젝트
Amazon SageMaker에서 머신러닝 모델을 구축하고 학습 및 배포하는 방법을 보여주는 Jupyter 노트북 모음입니다.
Hugging Face가 제공하는 Python 라이브러리로, 한 줄의 코드로 Hub의 공개 AI 데이터셋을 불러오고 Arrow 기반 도구로 효율적으로 전처리합니다.
머신러닝 실험, 데이터셋, 모델을 추적하고 시각화하며 관리하는 Python 라이브러리 및 호스팅 플랫폼입니다.
클라우드 또는 자체 호스팅으로 배포할 수 있는 오픈소스 AI 학습 추적·시각화 플랫폼으로, 지표와 하이퍼파라미터, 하드웨어 통계를 기록합니다.
자체 인프라나 클라우드 인프라에서 모델 학습부터 에이전트까지 ML 및 AI 워크플로를 파이프라인으로 구축하고 실행하는 오픈소스 Python 플랫폼입니다.