Dolfinx discontinous expression

[dokken]

That is one of the goals of dolfinx. For instance, you can use numba to create custom assemblers, use normal Python functions (lambda functions) to locate dofs, integrate expressions etc.