gpjax

Dependencies (35)

• beartype >0.16.1
• blackjax >=0.9.6
• equinox >=0.11.0
• ipykernel >=6.22.0
• ipython >=8.11.0
• ipywidgets >=8.0.5
• jax <0.11,>=0.5.0
• jaxlib <0.11,>=0.5.0

+ 27 more

Dependencies (36)

• beartype >0.16.1
• blackjax >=0.9.6
• equinox >=0.11.0
• ipykernel >=6.22.0
• ipython >=8.11.0
• ipywidgets >=8.0.5
• jax <0.11,>=0.5.0
• jaxlib <0.11,>=0.5.0

+ 28 more

Dependencies (35)

• beartype >0.16.1
• blackjax >=0.9.6
• equinox >=0.11.0
• ipykernel >=6.22.0
• ipython >=8.11.0
• ipywidgets >=8.0.5
• jax <0.11,>=0.5.0
• jaxlib <0.11,>=0.5.0

+ 27 more

Dependencies (35)

• beartype >0.16.1
• blackjax >=0.9.6
• equinox >=0.11.0
• ipykernel >=6.22.0
• ipython >=8.11.0
• ipywidgets >=8.0.5
• jax <0.10,>=0.5.0
• jaxlib <0.10,>=0.5.0

+ 27 more

Dependencies (35)

• beartype >0.16.1
• blackjax >=0.9.6
• equinox >=0.11.0
• ipykernel >=6.22.0
• ipython >=8.11.0
• ipywidgets >=8.0.5
• jax >=0.5.0
• jaxlib >=0.5.0

+ 27 more

Dependencies (31)

• beartype >0.16.1
• blackjax >=0.9.6
• flax >=0.12.2
• ipykernel >=6.22.0
• ipython >=8.11.0
• ipywidgets >=8.0.5
• jax >=0.5.0
• jaxlib >=0.5.0

+ 23 more

Dependencies (47)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.12.0

+ 39 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.12.0

+ 38 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.12.0

+ 38 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.12.0

+ 38 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.12.0

+ 38 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.12.0

+ 38 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.10.0

+ 38 more

Dependencies (46)

• absolufy-imports >=0.3.1
• autoflake
• beartype >0.16.1
• black
• blackjax >=0.9.6
• codespell >=2.2.4
• coverage >=7.2.2
• flax >=0.10.0

+ 38 more

Dependencies (10)

• beartype >0.16.1
• cola-ml >=0.0.7
• flax >=0.10.0
• jax >=0.5.0
• jaxlib >=0.5.0
• jaxtyping >0.2.10
• numpy >=2.0.0
• numpyro

+ 2 more

Dependencies (10)

• beartype >0.16.1
• cola-ml >=0.0.7
• flax >=0.10.0
• jax >=0.5.0
• jaxlib >=0.5.0
• jaxtyping >0.2.10
• numpy >=2.0.0
• numpyro

+ 2 more

Dependencies (10)

• beartype >0.16.1
• cola-ml >=0.0.7
• flax >=0.10.0
• jax >=0.5.0
• jaxlib >=0.5.0
• jaxtyping >0.2.10
• numpy >=2.0.0
• numpyro

+ 2 more

Dependencies (10)

• beartype >0.16.1
• cola-ml >=0.0.7
• flax >=0.10.0
• jax >=0.5.0
• jaxlib >=0.5.0
• jaxtyping >0.2.10
• numpy >=2.0.0
• optax >0.2.1

+ 2 more

Dependencies (10)

• beartype >0.16.1
• cola-ml >=0.0.7
• flax >=0.10.0
• jax >=0.5.0
• jaxlib >=0.5.0
• jaxtyping >0.2.10
• numpy >=2.0.0
• optax >0.2.1

+ 2 more

Dependencies (10)

• beartype >0.16.1
• cola-ml >=0.0.7
• flax >=0.10.0
• jax >=0.5.0
• jaxlib >=0.5.0
• jaxtyping >0.2.10
• numpy >=2.0.0
• optax >0.2.1

+ 2 more

Dependencies (10)

• beartype >0.16.1
• cola-ml ==0.0.5
• flax <0.10.0
• jax <0.4.28
• jaxlib <0.4.28
• jaxtyping >0.2.10
• numpy <2.0.0
• optax >0.2.1

+ 2 more

Dependencies (11)

• beartype >0.16.1
• cola-ml ==0.0.5
• flax <0.10.0
• jax <0.4.28
• jaxlib <0.4.28
• jaxopt ==0.8.2
• jaxtyping >0.2.10
• numpy <2.0.0

+ 3 more

Dependencies (11)

• beartype >0.16.1
• cola-ml ==0.0.5
• flax >=0.8.5
• jax <0.4.28
• jaxlib <0.4.28
• jaxopt ==0.8.2
• jaxtyping >0.2.10
• numpy <2.0.0

+ 3 more

Dependencies (11)

• beartype >0.16.1
• cola-ml ==0.0.5
• flax >=0.8.5
• jax <0.4.28
• jaxlib <0.4.28
• jaxopt ==0.8.2
• jaxtyping >0.2.10
• numpy <2.0.0

+ 3 more

Dependencies (11)

• beartype >0.16.1
• cola-ml ==0.0.5
• flax >=0.8.5
• jax <0.4.28
• jaxlib <0.4.28
• jaxopt ==0.8.2
• jaxtyping >0.2.10
• numpy <2.0.0

+ 3 more

Dependencies (11)

• beartype <0.17.0,>=0.16.1
• cola-ml ==0.0.5
• flax <0.9.0,>=0.8.4
• jax <0.4.28
• jaxlib <0.4.28
• jaxopt <0.9.0,>=0.8.3
• jaxtyping <0.3.0,>=0.2.10
• numpy <2.0.0

+ 3 more

Dependencies (11)

• beartype <0.17.0,>=0.16.2
• cola-ml <0.0.6,>=0.0.5
• jax >=0.4.16
• jaxlib >=0.4.16
• jaxopt <0.9.0,>=0.8.3
• jaxtyping <0.3.0,>=0.2.15
• optax <0.2.0,>=0.1.4
• orbax-checkpoint >=0.2.3

+ 3 more

Dependencies (10)

• beartype >=0.16.2,<0.17.0
• cola-ml >=0.0.5,<0.0.6
• jax >=0.4.16
• jaxlib >=0.4.16
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (11)

• beartype >=0.16.2,<0.17.0
• cola-ml >=0.0.5,<0.0.6
• jax >=0.4.10
• jaxlib >=0.4.10
• jaxopt >=0.8.2,<0.9.0
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3

+ 3 more

Dependencies (11)

• beartype >=0.16.2,<0.17.0
• cola-ml >=0.0.5,<0.0.6
• jax >=0.4.10
• jaxlib >=0.4.10
• jaxopt >=0.8.2,<0.9.0
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3

+ 3 more

Dependencies (10)

• beartype >=0.16.2,<0.17.0
• cola-ml >=0.0.5,<0.0.6
• jax >=0.4.10
• jaxlib >=0.4.10
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• cola-ml >=0.0.1,<0.0.2
• jax >=0.4.10
• jaxlib >=0.4.10
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• cola-ml >=0.0.1,<0.0.2
• jax >=0.4.10
• jaxlib >=0.4.10
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.10
• jaxlib >=0.4.10
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.3
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (10)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• plum-dispatch >=2.1.0,<3.0.0
• simple-pytree >=0.1.7,<0.2.0

+ 2 more

Dependencies (9)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• simple-pytree >=0.1.7,<0.2.0
• tensorflow-probability >=0.19.0,<0.20.0

+ 1 more

Dependencies (9)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• simple-pytree >=0.1.7,<0.2.0
• tensorflow-probability >=0.19.0,<0.20.0

+ 1 more

Dependencies (9)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib ==0.4.7
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• simple-pytree >=0.1.7,<0.2.0
• tensorflow-probability >=0.19.0,<0.20.0

+ 1 more

Dependencies (9)

• beartype >=0.13.1,<0.14.0
• jax >=0.4.1
• jaxlib >=0.4.6,<0.5.0
• jaxtyping >=0.2.15,<0.3.0
• optax >=0.1.4,<0.2.0
• orbax-checkpoint >=0.2.0,<0.3.0
• simple-pytree >=0.1.7,<0.2.0
• tensorflow-probability >=0.19.0,<0.20.0

+ 1 more

Dependencies (18)

• black
• deprecation
• distrax >=0.1.2
• flake8
• isort
• jax >=0.4.1
• jaxkern >=0.0.4
• jaxlib >=0.4.1

+ 10 more

Dependencies (18)

• black
• deprecation
• distrax >=0.1.2
• flake8
• isort
• jax >=0.4.1
• jaxkern >=0.0.4
• jaxlib >=0.4.1

+ 10 more

Dependencies (18)

• black
• deprecation
• distrax >=0.1.2
• flake8
• isort
• jax >=0.4.1
• jaxkern >=0.0.4
• jaxlib >=0.4.1

+ 10 more

Dependencies (19)

• black
• chex
• deprecation
• distrax >=0.1.2
• flake8
• isort
• jax >=0.4.1
• jaxkern

+ 11 more