Make FunctionSpace objects aware of whether they are axisymmetric

[btalamini]

When handed a FunctionSpace object, it can be useful to know if it does its calculus in axisymmetric coordinates or cartesian.

[tupek2]

we might consider moving away from namedtuples and use something like:

# data class
from chex._src.dataclass import dataclass
from chex._src import pytypes

@dataclass(frozen=True, mappable_dataclass=True)
class Polyhedral:
   interpolation : pytypes.ArrayDevice
   shapeGrad : pytypes.ArrayDevice
   weights : pytypes.ArrayDevice

Its easier to follow in code. No idea about the jit time. Also, that it is under chex._src is a bit odd.