search

liesel-devs / liesel

A probabilistic programming framework
https://liesel-project.org
MIT License
40 stars 2 forks source link

Make liesel compatible with jax v0.4.14 #73

Closed jobrachem closed 1 year ago

jobrachem commented 1 year ago

In Jax 0.4.14, the DeviceArray class has been removed (see changelog). As a result, liesel is incompatible with jax 0.4.14.

We use the DeviceArray class for type hints in three places in goose/summary_viz.py. These should be changed to jax.Array.