Update reference to jax.config

lnccbrown / HSSM

Development of HSSM package

Other

70 stars 10 forks source link

Update reference to jax.config #443

Closed digicosmos86 closed 1 month ago

digicosmos86 commented 1 month ago

JAX has marked from jax import config as deprecated as of 0.4.25 and this API was removed in 0.4.29. This affects the hssm.set_floatX API. We fix that in this PR