[MAINTAIN] Better equality check for jaxmarl specs

[sash-a]

Please describe what needs to be maintained?

This came up in #955. Unfortunately it's quite messy to check the equality of two jaxmarl spaces as they don't have custom __eq__ methods. So the current solution is to check that only the shapes match. This could cause possible bugs in the future so it would be great to find a better solution.