PMIx based wire-up for PyTorch
modify README
Mateusz Kapusta
(Jun 7, 2026, 11:04 PM +0200)
70cbdde1
4072e017
+9
-2
+9
-2
README.md
+9
-2
README.md
···
1
1
# Rixa
2
2
3
-
> ⚠️ **Experimental**
3
+
> [!WARNING]
4
4
> This library is in an experimental phase and may be buggy.
5
5
> Expect breaking changes between versions.
6
6
···
20
20
3. ABI Compatibility: Built on PMIx 5.0+ (ABI stable), ensuring portability across different MPI/PMIx versions.
21
21
22
22
## Installation
23
+
> [!NOTE]
24
+
> For detailed installation instructions, see the [wiki](https://github.com/wesenheit/rixa/wiki/Installation).
25
+
26
+
23
27
In order to install one needs to specify the version of the library. Currently two versions are supported, `pytorch` and `nvshmem`.
24
28
They can be easily installed with
25
29
```bash
···
29
33
# Install for NVSHMEM support
30
34
pip install "rixa[nvshmem]"
31
35
```
32
-
For detailed installation instructions, see the [Wiki](https://github.com/wesenheit/rixa/wiki).
33
36
34
37
## Usage (PyTorch)
38
+
> [!NOTE]
39
+
> For more use cases, see the [wiki](https://github.com/wesenheit/rixa/wiki/Quick-Start).
40
+
41
+
35
42
One can use `rixa` to start pytorch distributed job with one simple line
36
43
```python
37
44
import rixa