sagemaker-train has 1.4M monthly downloads and 2K GitHub stars. Version 1.15.0 supports Python 3.10+, first published in 2025.
It is actively maintained, last released in Jun 2026.
It is built on top of graphene, jinja2 and mlflow. It is used by 3 tracked packages, including sagemaker, sagemaker-serve and sagemaker-mlops.
GitHub topics include aws, mxnet and tensorflow.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
- graphene <4,>=3
- jinja2 <4.0,>=3.0
- mlflow <4.0.0,>=3.0.0
- nest-asyncio >=1.5.0
- paramiko >=2.11.0
- pydantic >=2.0.0
- pyyaml <7.0,>=6.0
- requests >=2.28.0
- sagemaker-core >=2.14.0
- sagemaker-mlflow <1.0.0,>=0.0.1
- tblib >=1.7.0
- typing-extensions >=4.9.0
15 optional dependencies
- graphene[test]
- httpx[rft-langchain]
- ipython[test]
- ipywidgets[notebook]
- matplotlib[notebook]
- mock[test]
- omegaconf[test]
- pandas[test]
- pydantic[test]
- pytest[test]
- pytest-cov[test]
- pytest-xdist[test]
- rich[notebook]
- scipy[test]
- strands-agents[rft-strands]