flax

c257e9e1 - added init_cache docstring example