flax

37a224e9 - [nnx] add gemma notebook