jax-ml/oryx

View on GitHub

Oryx is a library for probabilistic programming and deep learning built on top of Jax.

Stars: 328
Forks: 13
Open beginner issues: 0
Indexed issues: 17
Avg merge: 49m
Merged PRs (30d): 2

Dominant language: Python
License: Apache-2.0
Last GitHub push: Sep 10, 2026
Latest indexed: Sep 18, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: good first issue

Beginner friendly Everything

Open All

17 open issues indexed

Cannot import oryx due to jax AttributeError Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

jax-ml/oryx#108 · 1 reaction · Opened Jun 24, 2025
Bug with IteratedSigmoidCentered Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

jax-ml/oryx#102 · 2 comments · Opened Apr 21, 2025
Problem conditioning on `vmap` Poisson random variables Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

jax-ml/oryx#96 · 4 comments · 1 reaction · Opened Feb 3, 2025
Missing tag for 0.2.8 and 0.2.9 Open

Difficulty 3/5 1-2 days Newbie friendliness 35/100

jax-ml/oryx#95 · Opened Jan 14, 2025
[harvest] Document or fix that sow/reap doesn't work with keyword arguments Open

Difficulty 3/5 1-2 days Newbie friendliness 38/100

jax-ml/oryx#86 · Opened Nov 13, 2024
log_prob of blocked programs Open

Difficulty 4/5 3-5 days Newbie friendliness 38/100

jax-ml/oryx#81 · 2 comments · Opened Aug 29, 2024
Confusion about "append" mode in `sow` Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

jax-ml/oryx#78 · Opened Jun 17, 2024
Const sows dropped in transformations Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

jax-ml/oryx#75 · Opened Apr 30, 2024
Can't harvest-plant jitted scan in clobber mode Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

jax-ml/oryx#71 · Opened Mar 29, 2024
Can't plant into scan in append mode Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

jax-ml/oryx#70 · Opened Mar 29, 2024
Bad avals planted into remat Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

jax-ml/oryx#69 · Opened Mar 29, 2024
expit of logistic variable gives log_prob 0.0 Open

Difficulty 3/5 1-2 days Newbie friendliness 55/100

jax-ml/oryx#65 · Opened Feb 27, 2024
Pull harvest out into a separate pkg Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

jax-ml/oryx#57 · 6 comments · Opened Aug 21, 2023
Any insight into the dev model for `oryx`? Open

Difficulty 1/5 Under an hour Newbie friendliness 10/100

jax-ml/oryx#38 · 4 comments · Opened Feb 13, 2023
Fix the pjit support for oryx Open

Difficulty 5/5 Over a week Newbie friendliness 15/100

jax-ml/oryx#29 · 1 comment · Opened Jan 25, 2023
Specify Module's PyTree-Representation for jit/grad seperately. I.e. How to freeze state.variables Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

jax-ml/oryx#22 · Opened Oct 15, 2022
Add an example of integration with Blackjax? Open

enhancement good first issue

Difficulty 4/5 3-5 days Newbie friendliness 35/100

jax-ml/oryx#15 · 1 comment · Opened Sep 21, 2022

jax-ml/oryx

Get new issues in your inbox