[trax] Explicitly set `jax_pmap_shmap_merge=False`.

Remove more dtype canonicalization.

Configure tests to explicitly use jax_threefry_partitionable=False.

Automated Code Change

Avoid use of deprecated xla_bridge.get_backend().live_buffers()

remove deprecated assertDictContainsSubset

Future-proof calls to jnp.solve on batched 1D inputs.

disable certain tests in lax_numpy_test when numpy 2.0 is used

disable testOp when numpy 2.0 is installed

fix a few test bitwise op tests for np 2.0

skip testLinSpace and testLogSpace if numpy 2.0 is used

[numpy] Fix test failures under NumPy 2.0.

[numpy] Fix test failures under NumPy 2.0.

[scipy] Use NumPy functions rather than their deprecated aliases in SciPy's namespace.

Replace deprecated `jax.tree_*` functions with `jax.tree.*`