Update jaxlib requirement from <=0.4.14 to <=0.4.31 in /tests

Update jaxlib requirement from <=0.4.14 to <=0.4.31 in /tests #98