flax

d5b0605a - move functools import to cell