Add nodewise_{dot_product,max} to JIT discretization