The recommended dose of caffeine citrate for the treatment of neonatal apnea is typically 20 mg/kg as a loading dose, followed by a maintenance dose of 5-10 mg/kg/day. It is important to consult with a healthcare provider for the appropriate dosing regimen for each individual patient.