Clarification for LSTMCell Documentation

Hello, I was trying to understand the [LSTMCell](https://flax.readthedocs.io/en/v0.8.0/api_reference/flax.linen/_autosummary/flax.linen.LSTMCell.html) of Flax. The documentation for the ```__call__``` function says:

```
carry – the hidden state of the LSTM cell, initialized using LSTMCell.initialize_carry.
```

It thought it was weird that the cell state wasn't returned in addition with the hidden state. But in the [source code](https://flax.readthedocs.io/en/v0.8.0/_modules/flax/linen/recurrent.html#LSTMCell), ```initialize_carry``` seems to return a tuple containing the cell and the hidden states :

```python
return (c, h)
```

Additionally, the ```__call__``` function seems to also returns both the cell state and the hidden state in the carry:

```python
return (new_c, new_h), new_h
```

Did I misunderstand something? If not, should the documentation be updated to clarify that the carry includes both the cell state and the hidden state?

Anyway, thanks for the great library!