jax-ml / jax-ml/oryx

Bug with IteratedSigmoidCentered

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

Description

How to replicate:

```python
import oryx
import jax.numpy as jnp
oryx.bijectors.IteratedSigmoidCentered().forward(jnp.array([0., 0., 0.]))
```

Error message:

```
TypeError: abstract_eval_fun() missing 1 required keyword-only argument: 'debug_info'
```

I was wondering if I'm doing something wrong here, or this is just a bug with the bijector.

Thanks!

Contributor guide

Open the contributing guide

Research direction

Start by running the provided Python reproduction for oryx.bijectors.IteratedSigmoidCentered().forward with the shown JAX array and inspect the resulting abstract_eval_fun error. The work is done when this call completes without the missing debug_info TypeError and a regression test covers the reproduction.

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
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.