a2nfinance / detrain-python-lib Public

Notifications You must be signed in to change notification settings
Fork 0
Star 1

DeTrain primarily focuses on MP training, with two types: Tensor parallelism (TP) and Pipeline parallelism (PP).

1 star 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
detrain		detrain
tests		tests
.gitignore		.gitignore
README.md		README.md
README.rst		README.rst
pyproject.toml		pyproject.toml
setup.py		setup.py

Repository files navigation

Introduction

This package focuses on model parallelism training jobs with two types: tensor and pipeline. Source code is published on PyPI as DeTrain package

Commands

For local development: pip install -e .
To publish to pypi:
- python3 -m build
- python3 -m twine upload dist/*

About

DeTrain primarily focuses on MP training, with two types: Tensor parallelism (TP) and Pipeline parallelism (PP).

Report repository

Languages

Python 100.0%