Different boundary integral values between FENICS and JAX - trying to recreate boundary integral in JAX that is consistent with FENICS

Symfem should be able to provide arbitrary order basis functions.
For your test case, you are using Lagrange 1 and 2, which you can find at