jax

80b9bdc4 - cleanup, fix chunksize bug