Add JuMP function

[blegat]

According to https://github.com/jump-dev/JuMP.jl/issues/3980#issuecomment-2820913205, we should define set_forward_constraint_function etc.. that do the reshape and call moi_function and not do it in MOI.set like https://github.com/jump-dev/DiffOpt.jl/blob/43a82263118a69cc817042eabfe1d44c6e5ebedd/src/jump_moi_overloads.jl#L21