flax

d4653e55 - Merge pull request #5270 from samanklesaria:split_method