jax-ml / jax-ml/oryx

[harvest] Document or fix that sow/reap doesn't work with keyword arguments

Open
#86 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
328
Forks
13
Avg merge
49m
Merged PRs (30d)
2

Description

Reproducing example:

```
from oryx.core.interpreters.harvest import sow, reap
import jax

@jax.jit
def f(x):
sow(x, tag="d", name="x")

@jax.jit
def g(x):
return reap(f, tag="d")(x=x)

g(5) # {}
```

Contributor guide

Open the contributing guide

Research direction

Start with the sow and reap implementations in oryx/core/interpreters/harvest.py and reproduce the keyword-argument example from the issue. Determine whether the behavior should be fixed or documented, then verify that the chosen outcome explains or resolves g(x) returning {}.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Bug
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
38/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.