What version of DOLFINx are you running? The module basix.ufl was introduced after the v0.6.0 release.
basix.ufl