hawkinsp and fedjax authors Suppress some pytype errors related to improved JAX types under pytype.