File tree Expand file tree Collapse file tree 1 file changed +20
-1
lines changed Expand file tree Collapse file tree 1 file changed +20
-1
lines changed Original file line number Diff line number Diff line change 6
6
7
7
[ ![ integration tests] ( https://github.com/pytorch/torchtitan/actions/workflows/integration_test_8gpu.yaml/badge.svg?branch=main )] ( https://github.com/pytorch/torchtitan/actions/workflows/integration_test_8gpu.yaml?query=branch%3Amain )
8
8
[ ![ arXiv] ( https://img.shields.io/badge/arXiv-2410.06511-b31b1b.svg )] ( https://arxiv.org/abs/2410.06511 )
9
- [ ![ ICLR] ( https://img.shields.io/badge/ICLR-2025-blue .svg )] ( https://iclr.cc/virtual/2025/poster/29620 )
9
+ [ ![ ICLR] ( https://img.shields.io/badge/ICLR-2025-violet .svg )] ( https://iclr.cc/virtual/2025/poster/29620 )
10
10
[ ![ forum] ( https://img.shields.io/badge/pytorch-forum-DE3412.svg )] ( https://discuss.pytorch.org/c/distributed/torchtitan/44 )
11
11
[ ![ license] ( https://img.shields.io/badge/license-BSD_3--Clause-lightgrey.svg )] ( ./LICENSE )
12
+ [ ![ pip] ( https://img.shields.io/pypi/v/torchtitan?color=blue )] ( https://pypi.org/project/torchtitan/ )
13
+ [ ![ conda] ( https://img.shields.io/conda/vn/conda-forge/torchtitan?color=green )] ( https://anaconda.org/conda-forge/torchtitan )
14
+
12
15
13
16
</div >
14
17
@@ -86,6 +89,22 @@ You may want to see how the model is defined or how parallelism techniques are a
86
89
87
90
## Installation
88
91
92
+ ### Nightly
93
+
94
+ Coming soon.
95
+
96
+ ### Stable
97
+
98
+ Via pip:
99
+ ``` sh
100
+ pip install torchtitan
101
+ ```
102
+ Or via conda:
103
+ ``` sh
104
+ conda install conda-forge::torchtitan
105
+ ```
106
+ ### Sources
107
+
89
108
``` bash
90
109
git clone https://github.com/pytorch/torchtitan
91
110
cd torchtitan
You can’t perform that action at this time.
0 commit comments