Как использовать PYMC, PYMC-маркетинг и Numpyro вместе, когда им требуются противоречивые версии JAX? ⇐ Python

[Anonymous]

Я пытаюсь построить вероятностную модель на питоне, которая требует использования pymc , pymc-marketing и numpyro в одном и том же сценарии .

• . 64-битный < /strong> < /p>
  < /li>
  Установленные пакеты: < /p>
  

  Код: Выделить всё

  pymc

  (последний)

• Код: Выделить всё

  pymc-marketing

  (последний)

• Код: Выделить всё

  numpyro 0.18.0

• Код: Выделить всё

  jax

  (последний)

Когда я пытаюсь импортировать Numpyro после установки всего, я получаю:

Код: Выделить всё

ImportError: cannot import name 'pjit_p' from 'jax.experimental.pjit'

Это происходит потому, что:

[*] pymc и pymc-marketing требует jax (≥0.4.20+).
[*] numpyro 0.18.0 все еще импортирует PJIT_P , который был , который был , который был , который был , который был , который был , который был , который был . < /Li>
Понижение JAX до 0.4.14 Исправляет Numpyro, но ломает PYMC и PYMC-Marketing. < /Li>
< /ul>
Что я пробовал: < /h3>


< /ul>
Вопрос: < /h3>
Есть ли известный способ сделать PYMC, PYMC-маркетинг и Numpyro работать вместе в одной среде < /strong> с Python 3.12? Jax?>

Подробнее здесь: https://stackoverflow.com/questions/797 ... e-conflict