flax

77da09e9 - Merge pull request #3634 from chiamp:batch_apply