Attention for Inference Compilation