{"id":15710496,"url":"https://github.com/aws/sagemaker-mxnet-training-toolkit","last_synced_at":"2026-03-16T03:34:06.222Z","repository":{"id":57463628,"uuid":"118977299","full_name":"aws/sagemaker-mxnet-training-toolkit","owner":"aws","description":"Toolkit for running MXNet training scripts on SageMaker. Dockerfiles used for building SageMaker MXNet Containers are at https://github.com/aws/deep-learning-containers. ","archived":false,"fork":false,"pushed_at":"2025-02-11T01:28:54.000Z","size":13499,"stargazers_count":60,"open_issues_count":6,"forks_count":51,"subscribers_count":51,"default_branch":"master","last_synced_at":"2026-03-02T21:49:49.775Z","etag":null,"topics":["aws","docker","mxnet","sagemaker"],"latest_commit_sha":null,"homepage":"","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/aws.png","metadata":{"files":{"readme":"README.rst","changelog":"CHANGELOG.md","contributing":"CONTRIBUTING.md","funding":null,"license":"LICENSE","code_of_conduct":"CODE_OF_CONDUCT.md","threat_model":null,"audit":null,"citation":null,"codeowners":"CODEOWNERS","security":null,"support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2018-01-25T22:55:24.000Z","updated_at":"2025-02-11T01:28:58.000Z","dependencies_parsed_at":"2025-03-14T13:06:48.437Z","dependency_job_id":"672c46aa-75b9-4236-a6e0-9c87f6767807","html_url":"https://github.com/aws/sagemaker-mxnet-training-toolkit","commit_stats":null,"previous_names":["aws/sagemaker-mxnet-containers"],"tags_count":43,"template":false,"template_full_name":null,"purl":"pkg:github/aws/sagemaker-mxnet-training-toolkit","repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aws%2Fsagemaker-mxnet-training-toolkit","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aws%2Fsagemaker-mxnet-training-toolkit/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aws%2Fsagemaker-mxnet-training-toolkit/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aws%2Fsagemaker-mxnet-training-toolkit/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/aws","download_url":"https://codeload.github.com/aws/sagemaker-mxnet-training-toolkit/tar.gz/refs/heads/master","sbom_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/aws%2Fsagemaker-mxnet-training-toolkit/sbom","scorecard":null,"host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":286080680,"owners_count":30333443,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2026-03-10T05:25:20.737Z","status":"ssl_error","status_checked_at":"2026-03-10T05:25:17.430Z","response_time":106,"last_error":"SSL_read: unexpected eof while reading","robots_txt_status":"success","robots_txt_updated_at":"2025-07-24T06:49:26.215Z","robots_txt_url":"https://github.com/robots.txt","online":false,"can_crawl_api":true,"host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["aws","docker","mxnet","sagemaker"],"created_at":"2024-10-03T21:08:13.700Z","updated_at":"2026-03-10T12:31:25.725Z","avatar_url":"https://github.com/aws.png","language":"Python","funding_links":[],"categories":[],"sub_categories":[],"readme":"================================\nSageMaker MXNet Training Toolkit\n================================\n\nSageMaker MXNet Training Toolkit is an open-source library for using MXNet to train models on Amazon SageMaker.\nFor inference, see `SageMaker MXNet Inference Toolkit \u003chttps://github.com/aws/sagemaker-mxnet-serving-container\u003e`__.\nFor the Dockerfiles used for building SageMaker MXNet Containers, see `AWS Deep Learning Containers \u003chttps://github.com/aws/deep-learning-containers\u003e`__.\nFor information on running MXNet jobs on Amazon SageMaker, please refer to the `SageMaker Python SDK documentation \u003chttps://github.com/aws/sagemaker-python-sdk\u003e`__.\n\n\nContributing\n------------\n\nPlease read `CONTRIBUTING.md \u003chttps://github.com/aws/sagemaker-mxnet-training-toolkit/blob/master/CONTRIBUTING.md\u003e`__\nfor details on our code of conduct, and the process for submitting pull requests to us.\n\nTesting\n-------\n\nSet up a virtual environment for testing.\n~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~\n\nOne of the multiple ways to setup a virtual environment\n\n::\n\n   # use a package virtualenv\n   # create a virtualenv\n   virtualenv -p python3 \u003cname of env\u003e\n   # activate the virtualenv\n   source \u003cname of env\u003e/bin/activate\n\nInstall requirements\n~~~~~~~~~~~~~~~~~~~~\n\n::\n\n   pip install --upgrade .[test]\n\n\nLocal Test\n~~~~~~~~~~\n\nTo run specific test\n\n::\n\n   tox -- -k test/unit/test_training.py::test_train_for_distributed_scheduler\n\nTo run an entire file\n\n::\n\n   tox -- test/unit/test_training.py\n\nTo run all tests within a folder [e.g. integration/local/]\n\nNote: To run integration tests locally, one needs to build an image. To trigger image build, use `-B` flag.\n\n::\n\n   tox -- test/integration/local\n   \nYou can also run them in parallel:\n\n::\n\n   tox -- -n auto test/integration/local\n\nTo run for specific interpreter [Python environment], use the ``-e`` flag\n\n::\n\n   tox -e py37 -- test/unit/test_training.py\n\nRemote Test\n~~~~~~~~~~~\n\nMake sure to provide AWS account ID, Region, Docker base name \u0026 Tag.\nDocker Registry is composed of (aws_id, region)\nImage URI is composed of (docker_registry, docker_base_name, tag)\n\nResulting Image URI is composed as: ``{aws_id}.dkr.ecr.{region}.amazonaws.com/{docker_base_name}:{tag}``\n\n::\n\n    tox -- --aws-id \u003caws_id\u003e --region \u003cregion\u003e --docker-base-name \u003cdocker_base_name\u003e --tag \u003ctag\u003e test/integration/sagemaker\n\nFor more details, refer `conftest.py \u003ctest/conftest.py\u003e`_\n\nLicense\n-------\n\nSageMaker MXNet Training Toolkit is licensed under the Apache 2.0 License.\nIt is copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.\nThe license is available at: http://aws.amazon.com/apache2.0/\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Faws%2Fsagemaker-mxnet-training-toolkit","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Faws%2Fsagemaker-mxnet-training-toolkit","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Faws%2Fsagemaker-mxnet-training-toolkit/lists"}