As far as I can tell, you should use dolfinx.fem.Expression, which you give pre-determined points on the reference element. Then, in turn, you can evaluate this expression at the physical points in any cell by calling dolfinx.fem.Expression.eval(mesh, numpy_array_of_cell_indices)
as shown in: