jax-ml/jax

Support profiling without source code changes

Open

#20,293 opened on Mar 18, 2024

View on GitHubContributor guide
 (16 comments) (2 reactions) (0 assignees)Python (3,737 forks)github user discovery
enhancementgood first issue

Repository metrics

Stars
 (36,165 stars)
PR merge metrics
 (Avg merge 2d 19h) (439 merged PRs in 30d)

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

Contributor guide

Research direction
Examine how profiling is currently initiated in the codebase and implement an environment variable to enable profiling without source code changes. Focus on the entry points where JAX begins execution.
Tech stack
python
Domain
performancedeveloper experience
Issue type
Feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-2 days
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
Active
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
Clear
Prerequisites
Python
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
65