Use transformers to predict prime numbers.