jax-ml/jax

Support profiling without source code changes

Aperta

#20.293 aperta il 18 mar 2024

 (16 commenti) (2 reazioni) (0 assegnatari)Python (3738 fork)github user discovery
enhancementgood first issue

Metriche repository

Star
 (36.174 stelle)
Metriche merge PR
 (Merge medio 2g 19h) (439 PR mergiate in 30 g)

Descrizione

Currently adding profiling requires source-code changes which could be difficult in a large codebase.

Would be great if something like JAX_PROFILE=output.pb was supported to generate profiling from command line.

CC @hawkinsp

Guida contributor