We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 0cee7cf commit 911c082Copy full SHA for 911c082
2 files changed
src/art/megatron/shared.py
src/art/megatron/train.py
@@ -4,6 +4,14 @@
4
configure_megatron_runtime_env()
5
# isort: on
6
7
+"""Megatron training runtime and public worker API.
8
+
9
+Public cross-repo API consumed by serverless-training:
10
+- build_training_runtime
11
+- run_megatron_worker_loop
12
+- merge_lora_adapter
13
+"""
14
15
import gc
16
import importlib
17
import json
@@ -58,6 +66,17 @@
58
66
59
67
DEFAULT_MODEL_IDENTIFIER = "Qwen/Qwen3-30B-A3B-Instruct-2507"
60
68
69
+__all__ = [
70
+ "DEFAULT_MODEL_IDENTIFIER",
71
+ "TrainingRuntime",
72
+ "build_training_runtime",
73
+ "run_megatron_worker_loop",
74
+ "run_megatron_rl_job",
75
+ "run_megatron_sft_job",
76
+ "finalize_megatron_job",
77
+ "merge_lora_adapter",
78
+]
79
61
80
62
81
class TrainingRuntime(BaseModel):
63
82
model_config = ConfigDict(arbitrary_types_allowed=True)
0 commit comments