flax

d8708edd - Merge pull request #2999 from google:add-bidirectional-docs