flax

46579d13 - Call setup lazily.

Call setup lazily. We need to call setup() lazily rather than at object construction to avoid nasty exponential double-recursion overhead from repeated setup calls when methods are transformed by e.g. named_call for profiling.