flax
Fix setup in transformed class
#696
Merged

Loading