#1 firstElement and lastElement methods in JAX.NodeArray