A long time ago one had to use jax.core.symbolic_equal_dim to compare for equality dimensions that may include symbolic expressions (in presence of JAX polymorphism). This has not been
necessary for a while now, and we want to remove the jax.core.symbolic_equal_dim API.
Some projects seem to use core.definitely_equal for the same purpose. These changes can also be replaced by ==.
Replace jax.core.symbolic_equal_dim with ==
A long time ago one had to use
jax.core.symbolic_equal_dimto compare for equality dimensions that may include symbolic expressions (in presence of JAX polymorphism). This has not been necessary for a while now, and we want to remove thejax.core.symbolic_equal_dimAPI.Some projects seem to use
core.definitely_equalfor the same purpose. These changes can also be replaced by==.