Project description

JAX Synergistic Memory Inspector

In your JAX script:

from jax_smi import initialise_tracking
initialise_tracking()
# some computation...

Open a shell and run:

jax-smi

These details have not been verified by PyPI

View statistics for this project via Libraries.io, or by using our public dataset on Google BigQuery

1.0.3

Mar 10, 2023

1.0.2

Sep 3, 2022

1.0.1

Sep 3, 2022

This version

1.0.0

Sep 3, 2022

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Uploaded Sep 3, 2022 Source

Uploaded Sep 3, 2022 Python 3

Hashes for jax-smi-1.0.0.tar.gz
Algorithm	Hash digest
SHA256	`cf2cca6ae78e309dc39153fe104498540b153a6362da2e36a280687574776ade`
MD5	`b2af1849b7d088676aef0ad855eea1cc`
BLAKE2b-256	`7539d73cd6f92523ac0e659fcadee3c2b06b5fc89c612e153c7806a5d02fb46b`

Hashes for jax_smi-1.0.0-py3-none-any.whl
Algorithm	Hash digest
SHA256	`a4ca61686778ae93f992348d91b1a8e111b8bcf13ec71fd3e3a0c4b03b7baeda`
MD5	`7f757f5f37cebd1209e497305ff5afe8`
BLAKE2b-256	`d1f9a28cdf89f7297d6fdb00504232ef82a6cf17aae9fd0f54abf95804691665`