small error in kfs() function in parallel-kf-jax

The code currenty defines
```
def kfs(model, observations):
    return ks(model, *kf(model, observations, False))
```
but kf only takes 2 args. You should use
```
def kfs(model, observations):
    return ks(model, *kf(model, observations))
```