jax

2d2958c9 - fuse lazy expressions into eager computations