InfAlign: Inference-aware language model alignment