MultiHeadAttention documentation missing descriptions of `None` values for optional arguments
Open
- Dominant language
- Jupyter Notebook
- Stars
- 7.3k
- Forks
- 833
- Avg merge
- 5h 11m
- Merged PRs (30d)
- 5
Description
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`
Contributor guide
Assessment
This issue has not been assessed yet.