stormlog.jax.pprof_parser

Functions

parse_jax_memory_profile(file_path)

Parse a JAX .prof (gzipped pprof protobuf) using the official protobuf schema.

stormlog.jax.pprof_parser.parse_jax_memory_profile(file_path)

Parse a JAX .prof (gzipped pprof protobuf) using the official protobuf schema.

Parameters:

file_path (str)

Return type:

Dict[str, Any]