Can AI learn to do logic? We'll explore this question by teaching an AI to fill in gaps in mathematical proofs. Working with a database of over 40,000 proofs, we apply three different models to various prediction scenarios: a graph isomorphism network to predict a logical justification; an LSTM recurrent neural network to predict a logical statement; and an attention-based model to predict distance between statements within a proof.