Missing requirement for jax

LLNL / MuyGPyS

A fast, pure python implementation of the MuyGPs Gaussian process realization and training algorithm.

Other

23 stars 11 forks source link

Missing requirement for jax #163

Closed igoumiri closed 1 year ago

igoumiri commented 1 year ago

Ran into this when trying to use jax on pascal:

...
  File "/g/g90/goumiri1/src/monetgrams/venv/lib/python3.9/site-packages/MuyGPyS/_src/gp/kernels/jax.py", line 8, in <module>
    from tensorflow_probability.substrates import jax as tfp
ModuleNotFoundError: No module named 'tensorflow_probability'

Running pip install tensorflow_probability fixed it.

bwpriest commented 1 year ago

the readme takes care of these instructions: https://github.com/LLNL/MuyGPyS#jax-gpu-instructions