jax-ml/jax

Support profiling without source code changes

オープン

#20,293 opened on 2024/03/18

 (16 件のコメント) (2 件のリアクション) (0 人の担当者)Python (3,741 件のフォーク)github user discovery
enhancementgood first issue

Repository metrics

Stars
 (36,186 個のスター)
PR merge metrics
 (平均マージ 2d 19h) (30d で 439 merged PRs)

説明

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

コントリビューターガイド