1
- Distributed model parallel
2
- --------------------------
1
+ The SageMaker Distributed Model Parallel Library
2
+ ------------------------------------------------
3
3
4
4
The Amazon SageMaker distributed model parallel library is a model parallelism library for training
5
5
large deep learning models that were previously difficult to train due to GPU memory limitations.
@@ -9,49 +9,35 @@ allowing you to increase prediction accuracy by creating larger models with more
9
9
You can use the library to automatically partition your existing TensorFlow and PyTorch workloads
10
10
across multiple GPUs with minimal code changes. The library's API can be accessed through the Amazon SageMaker SDK.
11
11
12
- Use the following sections to learn more about the model parallelism and the library.
13
-
14
- Use with the SageMaker Python SDK
15
- =================================
16
-
17
- Use the following page to learn how to configure and enable distributed model parallel
18
- when you configure an Amazon SageMaker Python SDK `Estimator `.
12
+ See the following sections to learn more about the SageMaker model parallel library APIs.
19
13
20
14
.. toctree ::
21
- :maxdepth: 1
15
+ :maxdepth: 3
22
16
17
+ smp_versions/latest
23
18
smd_model_parallel_general
24
19
25
- API Documentation
26
- =================
27
-
28
- The library contains a Common API that is shared across frameworks, as well as APIs
29
- that are specific to supported frameworks, TensorFlow and PyTorch.
30
-
31
- Select a version to see the API documentation for version. To use the library, reference the
32
- **Common API ** documentation alongside the framework specific API documentation.
33
-
34
- .. toctree ::
35
- :maxdepth: 1
36
-
37
- smp_versions/latest.rst
38
- smp_versions/v1_3_0.rst
39
- smp_versions/v1_2_0.rst
40
- smp_versions/v1_1_0.rst
41
-
42
- It is recommended to use this documentation alongside `SageMaker Distributed Model Parallel
43
- <http://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel.html> `__ in the Amazon SageMaker
44
- developer guide. This developer guide documentation includes:
45
20
46
- - An overview of model parallelism and the library
47
- `core features <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-core-features.html >`__
48
- - Instructions on how to modify `TensorFlow
49
- <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-customize-training-script.html#model-parallel-customize-training-script-tf> `__
50
- and `PyTorch
51
- <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-customize-training-script.html#model-parallel-customize-training-script-pt> `__
52
- training scripts
53
- - `Configuration tips and pitfalls
54
- <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-customize-tips-pitfalls.html> `__
21
+ .. tip ::
22
+
23
+ We recommended using this API documentation with the conceptual guide at
24
+ `SageMaker's Distributed Model Parallel
25
+ <http://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel.html> `_
26
+ in the *Amazon SageMaker developer guide *. This developer guide documentation includes:
27
+
28
+ - An overview of model parallelism, and the library's
29
+ `core features <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-core-features.html >`_,
30
+ and `extended features for PyTorch <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-extended-features-pytorch.html >`_.
31
+ - Instructions on how to modify `TensorFlow
32
+ <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-customize-training-script-tf.html> `_
33
+ and `PyTorch
34
+ <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-customize-training-script-pt.html> `_
35
+ training scripts.
36
+ - Instructions on how to `run a distributed training job using the SageMaker Python SDK
37
+ and the SageMaker model parallel library
38
+ <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-sm-sdk.html> `_.
39
+ - `Configuration tips and pitfalls
40
+ <https://docs.aws.amazon.com/sagemaker/latest/dg/model-parallel-customize-tips-pitfalls.html> `_.
55
41
56
42
57
43
.. important ::
0 commit comments