[tool.poetry]
name = "oryx"
version = "0.2.4"
description = "Probabilistic programming and deep learning in JAX"
authors = ["Google LLC"]
license = "Apache 2.0"

[tool.poetry.dependencies]
python = "^3.7"
jax = "0.3.15"
jaxlib = "0.3.15"
tfp-nightly = {extras = ["jax"], version = "0.17.0.dev20220722"}


[tool.poetry.dev-dependencies]
inference-gym = "^0.0.4"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"

[tool.poetry]
name = "oryx"
version = "0.2.4"
description = "Probabilistic programming and deep learning in JAX"
authors = ["Google LLC"]
license = "Apache 2.0"

[tool.poetry.dependencies]
python = "^3.7"
jax = "0.3.20"
jaxlib = "0.3.20"
tensorflow-probability = {extras = ["jax"], version = "0.18.0"}


[tool.poetry.dev-dependencies]
inference-gym = "^0.0.4"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"

Update to latest version of jax and stable TFP
Update to latest version of jax and stable TFP
