Skip to content

Commit 82ff3ef

Browse files
authored
Support jax 0.7 by bumping numpyro (#325)
* support jax 0.7 by bumping numpyro * don't commit uv.lock
1 parent ff262cb commit 82ff3ef

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ Source = "https://github.com/BasisResearch/effectful"
4040
[project.optional-dependencies]
4141
torch = ["torch"]
4242
pyro = ["pyro-ppl>=1.9.1"]
43-
jax = ["jax<0.7"]
44-
numpyro = ["numpyro"]
43+
jax = ["jax"]
44+
numpyro = ["numpyro>=0.19"]
4545

4646
[dependency-groups]
4747
dev = [

0 commit comments

Comments
 (0)