In your JAX script:
from jax_smi import initialise_tracking
initialise_tracking()
# some computation...Open a shell and run:
jax-smi
forked from ayaka14732/jax-smi
-
Notifications
You must be signed in to change notification settings - Fork 0
JAX Synergistic Memory Inspector
License
neverix/jax-smi
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
JAX Synergistic Memory Inspector
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 100.0%