flax

7cbc3c13 - Merge pull request #3576 from chiamp:nnx_dynamic_slice