flax
cache cond/switch branch wrappers to avoid tracing cache misses
#5517

Open

cache cond/switch branch wrappers to avoid tracing cache misses #5517

mohsinm-dev wants to merge 1 commit into google:main from mohsinm-dev:fix-cond-tracing-cache-misses

cache cond/switch branch wrappers to avoid tracing cache misses

f9b94eb5

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone