flax

09537f49 - Merge pull request #5084 from samanklesaria:variable_eq