#1 optimize - removed function calling in ctor of JAX.Node