jax-ml/jax

Support profiling without source code changes

Ouverte

#20 293 ouverte le 18 mars 2024

 (16 commentaires) (2 réactions) (0 personne assignée)Python (3 745 forks)github user discovery
enhancementgood first issue

Métriques du dépôt

Stars
 (36 207 étoiles)
Métriques de merge PR
 (Merge moyen 2j 19h) (439 PRs mergées en 30 j)

Description

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

Guide contributeur