flax

bc93cb1d - Merge pull request #3814 from google:nnx-expose-sequential