flax

61280ffb - Merge pull request #1275 from jheek:cached-lifted-jit