flax

8525a9b7 - add typing overloads for variable