Towards generalizable neuro-symbolic reasoners