mirror of
https://github.com/Findus23/jax-array-info.git
synced 2024-09-19 15:53:47 +02:00
add installation
This commit is contained in:
parent
169e6a3fe6
commit
1a591faf79
1 changed files with 4 additions and 0 deletions
|
@ -2,6 +2,10 @@
|
|||
|
||||
This package contains two functions for debugging jax `Array`s:
|
||||
|
||||
```bash
|
||||
pip install git+https://github.com/Findus23/jax-array-info.git
|
||||
```
|
||||
|
||||
```python
|
||||
from jax_array_info import sharding_info, sharding_vis
|
||||
```
|
||||
|
|
Loading…
Reference in a new issue