dynamax
dynamax copied to clipboard
Tests failing due to incompatibility between TFP and JAX 0.7
Our tests are failing because of a compatibility issue with TFP and the newest version of JAX. See https://github.com/tensorflow/probability/issues/2009.