cortex-training 0.0.3__py3-none-any.whl

cortex_training/__init__.py

@@ -0,0 +1,49 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Public Python API for Cortex Training."""
+
+__version__ = "0.0.2"
+
+from . import wire
+from .client import ChunkGroupConflictError
+from .client import ChunkGroupError
+from .client import ChunkGroupRestartError
+from .client import CortexTrainingClient
+from .client import Hardware
+from .client import InferenceConfig
+from .client import JobType
+from .client import SubJobConfig
+from .client import TrainingConfig
+from .client import build_forward_backward_kwargs
+from .client import build_forward_backward_payload
+from .client import serialize_forward_backward_args
+
+__all__ = [
+    "CortexTrainingClient",
+    "build_forward_backward_kwargs",
+    "build_forward_backward_payload",
+    "ChunkGroupError",
+    "ChunkGroupRestartError",
+    "ChunkGroupConflictError",
+    "serialize_forward_backward_args",
+    "SubJobConfig",
+    "TrainingConfig",
+    "InferenceConfig",
+    "JobType",
+    "Hardware",
+    "wire",
+    "__version__",
+]

cortex_training/__main__.py

@@ -0,0 +1,20 @@
+# Copyright 2025 Snowflake Inc.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Module entry point for ``python -m cortex_training``."""
+
+from cortex_training.cli import main
+
+raise SystemExit(main())