stormlog.jax.attributed_viz

Stormlog-native memory visualisation for JAX (Directed Graph Dashboard).

Generates a directed call-graph using Graphviz, identical to go tool pprof’s graph view, and wraps it in a self-contained interactive HTML dashboard with Top Allocations tables and Summary stats.

Functions

`format_bytes`(b)	Format bytes into a human-readable string.
`render_jax_attributed_html`(profile_data[, ...])	Generate a self-contained HTML Dashboard from a parsed JAX pprof profile.

stormlog.jax.attributed_viz.format_bytes(b)[source]

Format bytes into a human-readable string.

Parameters:: b (float)
Return type:: str

stormlog.jax.attributed_viz.render_jax_attributed_html(profile_data, output_path='jax_memory_graph.html')[source]

Generate a self-contained HTML Dashboard from a parsed JAX pprof profile.

Parameters:

profile_data (Dict[str, Any])
output_path (str)

Return type:

str