The documentation for [linen.MultiHeadAttention](https://flax.readthedocs.io/en/latest/api_reference/flax.linen/layers.html#flax.linen.MultiHeadAttention) / [linen.MultiHeadDotProductAttention](https://flax.readthedocs.io/en/latest/api_reference/flax.linen/layers.html#flax.linen.MultiHeadDotProductAttention) is missing descriptions of what `None` entails for optional arguments. These include: - `qkv_features` - `out_features` - `deterministic` - `precision`
The documentation for linen.MultiHeadAttention / linen.MultiHeadDotProductAttention is missing descriptions of what
Noneentails for optional arguments. These include:qkv_featuresout_featuresdeterministicprecision