Let $\mathbb{F}$ be a field and W be the Weyl algebra, as the algebra over $\mathbb{F}$ generated by $a,b$ with relation $ab-ba=1$.
The description of simple modules over the Weyl algebra over algebraically closed fields of characteristic $p$ is known, they have dimension $p$ over $\mathbb F$ and they are in one to one correspondence with pairs of matrices of the form
$$M_a:=\begin{bmatrix} 0 & 0 & 0 & \cdots & \mu \\ 1 & 0 & 0 & \cdots & 0 \\ 0 & 1 & 0 & \cdots & 0 \\ \vdots & \ddots & \ddots & \ddots & \vdots \\ 0 & \cdots & 0 & 1 & 0 \end{bmatrix}, \quad M_b:=\begin{bmatrix} \lambda & p-1 & 0 & \cdots & 0 \\ 0 & \lambda & p-2 & \ddots & \vdots \\ 0 & 0 & \lambda & \ddots & 0 \\ \vdots & \vdots & \ddots & \ddots & 1 \\ 0 & 0 & \cdots & 0 & \lambda \end{bmatrix}$$
for $\mu,\lambda \in \mathbb{F}$.
Let $\mathbb{F}=\mathbb{Z}_p$ be the field of $p$ elements and assume that $M$ is a module such that the action of $a$ and $b$ is invertible. Then, condition $ab-ba=1$ implies that $\dim_{\mathbb{F}}(M)$ is a multiple of $p$. So, assume that $\dim_{\mathbb{F}}(M)=p$. Is it true that if $M$ is simple then $M$ is given by a pair of matrices as above?