The issue at https://github.com/tensorflow/probability/issues/1814 has been closed and numpy 2.0 offers many desired features.