AtlasMorph: Learning conditional deformable templates for brain MRI