JAX Synergistic Memory Inspector
Project description
JAX Synergistic Memory Inspector
Usage
In your JAX script:
from jax_smi import initialise_tracking
initialise_tracking()
# some computation...
Open a shell and run:
jax-smi
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
jax-smi-1.0.0.tar.gz
(5.8 kB
view hashes)