issues Search Results · repo:pyro-ppl/numpyro language:Python
Filter by
834 results
(69 ms)834 results
inpyro-ppl/numpyro (press backspace or delete to remove)Bug Description
Master depends on typing_extensions but does not install it. I think the dependency was introduced in #2028.
CI didn t catch this because the test extra has mypy.
Steps to Reproduce ...
bug
tare
- Opened 2 days ago
- #2053
import numpyro fails with jax 0.7.0 with the following error
import numpyro
ImportError: cannot import name pjit_p from jax.experimental.pjit (/Users/damon/Documents/GitHub/stf-pyrenew-added-value/.venv/lib/python3.13/site-packages/jax/experimental/pjit.py) ...
bug
damonbayer
- 1
- Opened 3 days ago
- #2051
Gradient clipping in numpyro
numpyro seems to use elementwise clipping when using the ClippedAdam class
https://github.com/pyro-ppl/numpyro/blob/6fd6045a3df94e7dbb61620615920f6c7c505c89/numpyro/optim.py#L204-L211 ...
enhancement
SamuelBrand1
- 1
- Opened 9 days ago
- #2047
Feature Summary
Add a divider as a parameter in random_nnx_module().
Why is this needed?
Currently, the default divider / is used here:
https://github.com/pyro-ppl/numpyro/blob/6fd6045a3df94e7dbb61620615920f6c7c505c89/numpyro/contrib/module.py#L565 ...
enhancement
markean
- Opened 10 days ago
- #2045
Feature Summary
It would be great to cut a patch release :)
Why is this needed?
I wanna make sure I have the fix in https://github.com/pyro-ppl/numpyro/pull/2024 for production models (installing from ...
enhancement
juanitorduz
- 5
- Opened 11 days ago
- #2044
Feature Summary
Currently, MixtureGeneral supports a custom support argument, as discussed in this issue comment. It would be helpful to
extend similar functionality to TransformedDistribution.
Why is ...
enhancement
Qazalbash
- 6
- Opened 12 days ago
- #2043
Bug Description
The WishartCholesky distribution accepts scale_matrix, rate_matrix, and scale_tril as array-type arguments, as seen in
the distribution s argument constraints. However, the infer_shapes ...
bug
- 6
- Opened 21 days ago
- #2041
Bug Description
The example notebook for Text-Based Ideal Points uses an old version of numpyro (v0.10.1) which doesn t work anymore
because the jax.util module imported by numpyro v0.10.1 has now been ...
bug
axelverrier
- Opened on Jun 24
- #2040
Feature Summary
Add a parameter called initial_value to RecursiveLinearTransform.
Why is this needed?
When using RecursiveLinearTransform for modeling structural time series, I sometimes run into limitations ...
enhancement
tingiskhan
- 1
- Opened on Jun 15
- #2037
Bug Description
In the PR, adding type hints to the distributions module https://github.com/pyro-ppl/numpyro/pull/2032, the CI did not
check for consistency, and we have many lint errors.
Steps to Reproduce ...
bug
- 2
- Opened on Jun 7
- #2036
Learn how you can use GitHub Issues to plan and track your work.
Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub IssuesProTip!
Restrict your search to the title by using the in:title qualifier.Learn how you can use GitHub Issues to plan and track your work.
Save views for sprints, backlogs, teams, or releases. Rank, sort, and filter issues to suit the occasion. The possibilities are endless.Learn more about GitHub IssuesProTip!
Restrict your search to the title by using the in:title qualifier.