flax

f99b0267 - Improve docs related with dataclasses