flax

f043c5e1 - add benchmark on state traversal, and a readme