Auto-differentiable Ensemble Kalman Filters