flax

70d845a8 - Remove pytype generic workaround