mirror/jax-array-info
1
0
Fork 0
mirror of https://github.com/Findus23/jax-array-info.git synced 2024-09-19 15:53:47 +02:00
Code Releases Activity

add installation

Browse source
This commit is contained in:
Lukas Winkler 2023-09-04 16:26:03 +02:00
parent 169e6a3fe6
commit 1a591faf79
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
README.md
View file

@ -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
```